logo
Browse Source

No need to load pretrained backbone

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
Jael Gu 2 years ago
parent
commit
224d76d379
  1. 2
      isc.py

2
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()

Loading…
Cancel
Save