diff --git a/train/train_food_classifier.py b/train/train_food_classifier.py index e736e8e..968eaf3 100644 --- a/train/train_food_classifier.py +++ b/train/train_food_classifier.py @@ -199,6 +199,13 @@ if __name__ == '__main__': if val_acc > best_val_acc: best_val_acc = val_acc best_model_path = settings.BEST_MODEL_PATH + + # 检查并创建模型保存目录 + model_dir = os.path.dirname(best_model_path) + if not os.path.exists(model_dir): + os.makedirs(model_dir) + print(f"创建模型保存目录: {model_dir}") + torch.save(model.state_dict(), settings.BEST_MODEL_PATH) print(f'保存最佳模型,验证准确率: {best_val_acc:.2f}%')