add save_model

Signed-off-by: gexy5 <xinyu.ge@zilliz.com>
2 years ago · cb951f924b
1 changed files with 4 additions and 3 deletions
--- a/timm_image.py
+++ b/timm_image.py
@ -100,6 +100,7 @@ class TimmImage(NNOperator):
 #     op = TimmImage('resnet50')
 #     out = op(img)
 #     print(out)
 #    op.model = torch.jit.load('resnet50.pt')
 #    out2 = op(img)
 #    print(out2)
 #
 #     op.model = torch.jit.load('resnet50.pt')
 #     out2 = op(img)
 #     print(out2)