每次训练完成后，保存训练结果。

2025-09-10 15:32:21 +08:00
parent 5a3fc59573
commit 97401e908a
2 changed files with 77 additions and 2 deletions
@@ -15,16 +15,17 @@ 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', '04')
+MODEL_DIR = os.path.join(BASE_DIR, 'model', '05')
 BEST_MODEL_PATH = os.path.join(MODEL_DIR, 'best_food_model.pth')
 TRAINING_CURVES_PATH = os.path.join(MODEL_DIR, 'training_curves.png')
+TRAINING_RESULTS_PATH = os.path.join(MODEL_DIR, 'training_results.txt')

 # 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
+NUM_EPOCHS = 13
 BATCH_SIZE = 32
 LEARNING_RATE = 0.001
 WEIGHT_DECAY = 1e-4