diff --git a/app/yolov5/train_server.py b/app/yolov5/train_server.py index fe799d9..b913f63 100644 --- a/app/yolov5/train_server.py +++ b/app/yolov5/train_server.py @@ -106,7 +106,7 @@ def train(hyp, opt, device, data_list,id,callbacks): # hyp is path/to/hyp.yaml (wd.parent if evolve else wd).mkdir(parents=True, exist_ok=True) # make dir # 获取版本号,项目名称 #version, pro_name = get_version_name(pro) - best = Path(opt.savemodel)#w + pro_name + '_' + 'R-ODY' + version + '.pt' + best = opt.savemodel#w + pro_name + '_' + 'R-ODY' + version + '.pt' #last, best = w / 'last.pt', w / 'best.pt' # Hyperparameters