From bb066c330cc89ade7fb064e79caf311b0e932edf Mon Sep 17 00:00:00 2001 From: Jael Gu Date: Fri, 10 Feb 2023 17:45:55 +0800 Subject: [PATCH] Update timm backbone Signed-off-by: Jael Gu --- isc.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/isc.py b/isc.py index 0253e11..f9a3d2e 100644 --- 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):