From 541f59afb1068cad39bb0601832fe3830db50441 Mon Sep 17 00:00:00 2001
From: zilliz <zilliz@zillizdeMacBook-Pro.local>
Date: Fri, 17 Dec 2021 10:04:54 +0800
Subject: [PATCH] fix embedding dimension problem

---
 pytorch/model.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/pytorch/model.py b/pytorch/model.py
index ea0e408..6a97391 100644
--- a/pytorch/model.py
+++ b/pytorch/model.py
@@ -28,6 +28,7 @@ class Model():
         super().__init__()
         model_func = getattr(torchvision.models, model_name)
         self._model = model_func(pretrained=True)
+        self._model.fc = torch.nn.Identify()
         self._model.eval()
 
     def __call__(self, img_tensor: torch.Tensor):