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':