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)