settings.py中增加推理模型路径。

2025-09-10 15:12:45 +08:00
parent 5f62c6e2e0
commit 2412c26342
2 changed files with 5 additions and 1 deletions
@@ -19,6 +19,9 @@ MODEL_DIR = os.path.join(BASE_DIR, 'model', '03')
 BEST_MODEL_PATH = os.path.join(MODEL_DIR, 'best_food_model.pth')
 TRAINING_CURVES_PATH = os.path.join(MODEL_DIR, 'training_curves.png')

+# INFERENCE_BEST_MODEL_PATH = os.path.join(BASE_DIR, 'model', '03','best_food_model.pth')
+INFERENCE_BEST_MODEL_PATH = BEST_MODEL_PATH
+
 # 训练参数
 # NUM_EPOCHS = 100
 NUM_EPOCHS = 3