Update

Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>

resnet_image_embedding.py

@@ -31,7 +31,7 @@ class ResnetImageEmbedding(Operator):
         sys.path.append(str(Path(__file__).parent))
         if framework == 'pytorch':
             from pytorch.model import Model
-        self.model = Model(model_name)
+        self.model = Model(model_name, weights_path)
 
     def __call__(self, img_tensor: torch.Tensor) -> NamedTuple('Outputs', [('feature_vector', numpy.ndarray)]):
         embedding = self.model(img_tensor)