在保存模型文件之前，先检查路径是否存在。

2025-09-10 15:05:08 +08:00
parent 06fead1ab3
commit 5f62c6e2e0
1 changed files with 7 additions and 0 deletions
@@ -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}%')