From 9c6a43c18475693c02c4666572c9bbe12505018c Mon Sep 17 00:00:00 2001
From: Jael Gu <mengjia.gu@zilliz.com>
Date: Tue, 27 Dec 2022 16:35:36 +0800
Subject: [PATCH] Debug to support triton

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
---
 isc.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/isc.py b/isc.py
index 7908977..f5b999f 100644
--- a/isc.py
+++ b/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':