Update

Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
4 years ago · 6308ec7a76
1 changed files with 1 additions and 1 deletions
--- 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':