diff --git a/settings/settings.py b/settings/settings.py index 8e9337c..489c792 100644 --- a/settings/settings.py +++ b/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') diff --git a/train/train_food_classifier.py b/train/train_food_classifier.py index 968eaf3..7cdfb6b 100644 --- a/train/train_food_classifier.py +++ b/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最终结果:')