Browse Source
Update timm backbone
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
1 changed files with
1 additions and
2 deletions
-
isc.py
|
|
@ -48,9 +48,8 @@ _ = sys.modules[__name__] |
|
|
class Model: |
|
|
class Model: |
|
|
def __init__(self, timm_backbone, checkpoint_path, device): |
|
|
def __init__(self, timm_backbone, checkpoint_path, device): |
|
|
self.device = 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, |
|
|
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() |
|
|
self.model.eval() |
|
|
|
|
|
|
|
|
def __call__(self, x): |
|
|
def __call__(self, x): |
|
|
|
