Update timm backbone

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
3 years ago · bb066c330c
1 changed files with 1 additions and 2 deletions
--- a/isc.py
+++ b/isc.py
@ -48,9 +48,8 @@ _ = sys.modules[__name__]
 class Model:
    def __init__(self, timm_backbone, checkpoint_path, device):
        self.device = device
-        self.backbone = create_model(timm_backbone, features_only=True, pretrained=False)
        self.model = isc.create_model(pretrained=True, checkpoint_path=checkpoint_path, device=self.device,
-                                      backbone=self.backbone, p=1.0, eval_p=1.0)
+                                      timm_backbone=timm_backbone, p=1.0, eval_p=1.0)
        self.model.eval()

    def __call__(self, x):