Browse Source
Update
Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
training
1 changed files with
1 additions and
1 deletions
-
resnet_image_embedding.py
|
|
@ -26,7 +26,7 @@ class ResnetImageEmbedding(Operator): |
|
|
""" |
|
|
""" |
|
|
PyTorch model for image embedding. |
|
|
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__() |
|
|
super().__init__() |
|
|
sys.path.append(str(Path(__file__).parent)) |
|
|
sys.path.append(str(Path(__file__).parent)) |
|
|
if framework == 'pytorch': |
|
|
if framework == 'pytorch': |
|
|
|
