logo
Browse Source

Debug to support triton

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
Jael Gu 2 years ago
parent
commit
9c6a43c184
  1. 3
      isc.py

3
isc.py

@ -114,7 +114,8 @@ class Isc(NNOperator):
@property
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'):
if path == 'default':

Loading…
Cancel
Save