优化训练过程

app/service/project_train_service.py

@@ -14,12 +14,11 @@ from app.websocket.web_socket_server import room_manager
 from app.util.csv_utils import read_csv
 from app.common.redis_cli import redis_conn
 
-from sqlalchemy.orm import Session
-from typing import List
-from fastapi import UploadFile
 import yaml
 import subprocess
-import torch
+from typing import List
+from fastapi import UploadFile
+from sqlalchemy.orm import Session
 
 
 def add_project(info: ProjectInfoIn, session: Session, user_id: int):