diff --git a/app/yolov5/train_server.py b/app/yolov5/train_server.py index ab5ac4e..0146c60 100644 --- a/app/yolov5/train_server.py +++ b/app/yolov5/train_server.py @@ -63,8 +63,8 @@ from app.schemas.TrainResult import Report, ProcessValueList from app.controller.AlgorithmController import algorithm_process_value_websocket from app.configs import global_var from app.utils.websocket_tool import manager -LOCAL_RANK = int(os.getenv('LOCAL_RANK', -1)) # https://pytorch.org/docs/stable/elastic/run.html -RANK = int(os.getenv('RANK', -1)) +LOCAL_RANK = -1 #int(os.getenv('LOCAL_RANK', -1)) # https://pytorch.org/docs/stable/elastic/run.html +RANK = -1 #int(os.getenv('RANK', -1)) WORLD_SIZE = int(os.getenv('WORLD_SIZE', 1)) #重写coco128.yaml,改成训练的数据路径和类别名称