修改配置

This commit is contained in:
2025-10-09 09:59:36 +08:00
parent f3e3bd048a
commit a14aa7f8e1
4 changed files with 4 additions and 3 deletions
+1 -1
View File
@@ -477,7 +477,7 @@ class FAISSSearcher:
def main():
"""主函数"""
# 配置参数
MODEL_PATH = "../model/embedding_20250929_151102/best_embedding_model.pth"
MODEL_PATH = "../model/embedding_20250930_102826/best_embedding_model.pth"
TRAIN_DIR = "../dataset/train"
OUTPUT_DIR = "faiss_index"
BATCH_SIZE = 16