src/models.py [142:152]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
                wpath = os.path.join(config.preTrained[self.net_idx], f"{self.name}.weights")
                if os.path.exists(wpath):
                    print("loading pretrained weights")
                    self.load_weights(wpath, device)
                else:
                    print(f"WARNING pretrained weights not found: {wpath}")
                opath = wpath = os.path.join(config.preTrained[self.net_idx], f"{self.name}.optimizer")
                if optimizer is not None and os.path.exists(opath):
                    self.load_optimizer(opath, optimizer, device)
            return 0
        ckpt_path = ckpts[-1]
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



src/models.py [319:329]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
                wpath = os.path.join(config.preTrained[self.net_idx], f"{self.name}.weights")
                if os.path.exists(wpath):
                    print("loading pretrained weights")
                    self.load_weights(wpath, device)
                else:
                    print(f"WARNING pretrained weights not found: {wpath}")
                opath = wpath = os.path.join(config.preTrained[self.net_idx], f"{self.name}.optimizer")
                if optimizer is not None and os.path.exists(opath):
                    self.load_optimizer(opath, optimizer, device)
            return 0
        ckpt_path = ckpts[-1]
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -