训练指标曲线图配置成setting文件中的变量。

This commit is contained in:
zhanghuan
2025-09-10 15:19:29 +08:00
parent 2412c26342
commit 5a3fc59573
2 changed files with 2 additions and 2 deletions
+1 -1
View File
@@ -15,7 +15,7 @@ VAL_DATA_DIR = os.path.join(DATASET_DIR, 'val')
TEST_DATA_DIR = os.path.join(DATASET_DIR, 'test')
# 模型保存路径
MODEL_DIR = os.path.join(BASE_DIR, 'model', '03')
MODEL_DIR = os.path.join(BASE_DIR, 'model', '04')
BEST_MODEL_PATH = os.path.join(MODEL_DIR, 'best_food_model.pth')
TRAINING_CURVES_PATH = os.path.join(MODEL_DIR, 'training_curves.png')