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

settings/settings.py

@@ -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')
 

train/train_food_classifier.py

@@ -246,7 +246,7 @@ if __name__ == '__main__':
     plt.grid(True)
 
     plt.tight_layout()
-    plt.savefig('../model/02/training_curves.png', dpi=300, bbox_inches='tight')
+    plt.savefig(settings.TRAINING_CURVES_PATH, dpi=300, bbox_inches='tight')
     plt.show()
 
     print(f'\n最终结果:')