From 6308ec7a76fac69e8d9396299335e3eee61f81c4 Mon Sep 17 00:00:00 2001 From: shiyu22 Date: Wed, 15 Dec 2021 21:19:26 +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 d10f224..1c0a8d5 100644 --- a/resnet_image_embedding.py +++ b/resnet_image_embedding.py @@ -26,7 +26,7 @@ class ResnetImageEmbedding(Operator): """ PyTorch model for image embedding. """ - def __init__(self, model_name: str, framework: str = 'pytorch') -> None: + def __init__(self, model_name: str, framework: str = 'pytorch', weights_path: str = None) -> None: super().__init__() sys.path.append(str(Path(__file__).parent)) if framework == 'pytorch':