logo
Browse Source

add save_model

Signed-off-by: gexy5 <xinyu.ge@zilliz.com>
main
gexy5 2 years ago
parent
commit
cb951f924b
  1. 7
      timm_image.py

7
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)

Loading…
Cancel
Save