diff --git a/isc.py b/isc.py index af85702..9c1d6d8 100644 --- a/isc.py +++ b/isc.py @@ -56,7 +56,7 @@ class Isc(NNOperator): if checkpoint_path is None: checkpoint_path = os.path.join(str(Path(__file__).parent), 'checkpoints', timm_backbone + '.pth') - backbone = timm.create_model(timm_backbone, features_only=True, pretrained=True) + backbone = timm.create_model(timm_backbone, features_only=True, pretrained=False) self.model = isc.create_model(pretrained=True, checkpoint_path=checkpoint_path, device=self.device, backbone=backbone, p=3.0, eval_p=1.0) self.model.eval()