From 766b44c29e3ac3fdb4f6d1d2a825ddc0cecec616 Mon Sep 17 00:00:00 2001 From: shiyu22 Date: Wed, 15 Dec 2021 21:20:37 +0800 Subject: [PATCH] Update Signed-off-by: shiyu22 --- resnet_image_embedding.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/resnet_image_embedding.py b/resnet_image_embedding.py index 1c0a8d5..5b7cdc9 100644 --- a/resnet_image_embedding.py +++ b/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)