增加注释，去掉dropout

net/resnet_embedding.py

@@ -42,11 +42,17 @@ class ResNet50EmbeddingNet(nn.Module):
         self.backbone.fc = nn.Identity()
         
         # 添加embedding层
+        # self.embedding_layer = nn.Sequential(
+        #     nn.Linear(backbone_output_dim, embedding_dim),
+        #     nn.BatchNorm1d(embedding_dim),
+        #     nn.ReLU(inplace=True),
+        #     nn.Dropout(0.2)
+        # )
+        # 暂时不用加Dropout
         self.embedding_layer = nn.Sequential(
             nn.Linear(backbone_output_dim, embedding_dim),
             nn.BatchNorm1d(embedding_dim),
-            nn.ReLU(inplace=True),
-            nn.Dropout(0.2)
+            nn.ReLU(inplace=True)
         )
         
         # 图片预处理变换（用于推理）