From cb951f924b14dcdc74918105df359e31bad7e821 Mon Sep 17 00:00:00 2001
From: gexy5 <xinyu.ge@zilliz.com>
Date: Thu, 2 Jun 2022 10:26:34 +0800
Subject: [PATCH] add save_model

Signed-off-by: gexy5 <xinyu.ge@zilliz.com>
---
 timm_image.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/timm_image.py b/timm_image.py
index 2683b7c..1b80e55 100644
--- 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)