From b6f7a04d16bb6986bac27692c53ef15c532a5aa1 Mon Sep 17 00:00:00 2001 From: JIAKUNHAO Date: Wed, 23 Nov 2022 21:12:25 +0800 Subject: [PATCH] =?UTF-8?q?GPU=E9=80=89=E6=8B=A9debug-6?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/yolov5/train_server.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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,改成训练的数据路径和类别名称