现在统一使用Torch中的interpolate这个缩放方法，不用torchvision中的transforms的Resize了，应该就可以了。

2025-09-11 17:56:25 +08:00
parent e1cb936e32
commit da67cb9f4c
2 changed files with 15 additions and 5 deletions
@@ -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', '09')
+MODEL_DIR = os.path.join(BASE_DIR, 'model', '10')
 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')