Browse Source
No need to load pretrained backbone
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
1 changed files with
1 additions and
1 deletions
-
isc.py
|
|
@ -56,7 +56,7 @@ class Isc(NNOperator): |
|
|
if checkpoint_path is None: |
|
|
if checkpoint_path is None: |
|
|
checkpoint_path = os.path.join(str(Path(__file__).parent), 'checkpoints', timm_backbone + '.pth') |
|
|
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, |
|
|
self.model = isc.create_model(pretrained=True, checkpoint_path=checkpoint_path, device=self.device, |
|
|
backbone=backbone, p=3.0, eval_p=1.0) |
|
|
backbone=backbone, p=3.0, eval_p=1.0) |
|
|
self.model.eval() |
|
|
self.model.eval() |
|
|
|
