调整模型路径。

2025-10-16 14:41:59 +08:00
parent 3fd23dcc7d
commit e74c42b5d4
3 changed files with 6 additions and 3 deletions
@@ -17,8 +17,9 @@ def main():
    # 1. 加载训练好的embedding模型权重
    # base_model = create_resnet50_embedding(embedding_dim=512, pretrained=True)
    base_model = create_mobile_resnet50_embedding(embedding_dim=512, pretrained=True)
-    model_path = "../model/embedding_20250930_102826/best_embedding_model.pth"
-    
+    # model_path = "../model/embedding_20250930_102826/best_embedding_model.pth"
+    model_path = "../model/embedding_20251011_133653/best_embedding_model.pth"
+
    if not os.path.exists(model_path):
        print(f"错误：模型文件不存在 {model_path}")
        return
@@ -85,7 +86,7 @@ def main():
        traced_model = torch.jit.trace(mobile_wrapper, single_input)
        
        # 保存模型
-        output_path = "../model/embedding_20250930_102826/best_embedding_model_mobile.pt"
+        output_path = "../model/embedding_20251011_133653/best_embedding_model_mobile.pt"
        traced_model.save(output_path)
        print(f"✓ TorchScript模型保存成功: {output_path}")