xkrs_manan/RODY
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

GPU选择debug-3

Browse Source
This commit is contained in:
JIAKUNHAO 2022-11-23 20:52:25 +08:00
parent 8b428437e9
commit 3da6f21b44
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
app/controller/AlgorithmController.py
View File

@ -305,7 +305,7 @@ def train_R0DY(params_str, id):
from app.yolov5.train_server import train_start
params = TrainParams()
params.read_from_str(params_str)
print(params.get('device').default)
print(params.get('device').value)
data_list = file_tool.get_file(ori_path=params.get('DatasetDir').value, type_list=params.get('CLASS_NAMES').value)
weights = params.get('resumeModPath').value # 初始化模型绝对路径
img_size = params.get('img_size').value

1
app/yolov5/train_server.py
View File

@ -608,6 +608,7 @@ def main(opt,data_list,id,callbacks=Callbacks()):
opt.save_dir = str(increment_path(Path(opt.project) / opt.name, exist_ok=opt.exist_ok))
# DDP mode
print('\n','opt.device:',opt.device,'\n')
device = select_device(opt.device, batch_size=opt.batch_size)
print('\n','device:',device,'\n')
# if LOCAL_RANK != -1: