Browse Source
Debug to support triton
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
1 changed files with
2 additions and
1 deletions
-
isc.py
|
@ -114,7 +114,8 @@ class Isc(NNOperator): |
|
|
|
|
|
|
|
|
@property |
|
|
@property |
|
|
def backbone(self): |
|
|
def backbone(self): |
|
|
return self.model.backbone |
|
|
|
|
|
|
|
|
backbone = timm.create_model(self.timm_backbone, features_only=True, pretrained=False) |
|
|
|
|
|
return backbone |
|
|
|
|
|
|
|
|
def save_model(self, format: str = 'pytorch', path: str = 'default'): |
|
|
def save_model(self, format: str = 'pytorch', path: str = 'default'): |
|
|
if path == 'default': |
|
|
if path == 'default': |
|
|