From 224d76d37938c9160b238db91da8910f459394f6 Mon Sep 17 00:00:00 2001 From: Jael Gu Date: Mon, 28 Nov 2022 15:28:31 +0800 Subject: [PATCH] No need to load pretrained backbone Signed-off-by: Jael Gu --- isc.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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()