From 950e29eb4932fa2b603460fbd24184a2e20be09b Mon Sep 17 00:00:00 2001 From: JIAKUNHAO Date: Fri, 18 Nov 2022 15:20:33 +0800 Subject: [PATCH] =?UTF-8?q?=E5=AE=9E=E6=97=B6=E8=BF=94=E5=9B=9E=E8=AE=AD?= =?UTF-8?q?=E7=BB=83loss=20report?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/yolov5/train_server.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/app/yolov5/train_server.py b/app/yolov5/train_server.py index 9a625a7..1b42781 100644 --- a/app/yolov5/train_server.py +++ b/app/yolov5/train_server.py @@ -400,6 +400,7 @@ def train(hyp, opt, device, data_list,id,callbacks): # hyp is path/to/hyp.yaml # Backward scaler.scale(loss).backward() + report_cellback(epoch, epochs, float(loss)) # Optimize - https://pytorch.org/docs/master/notes/amp_examples.html if ni - last_opt_step >= accumulate: @@ -484,7 +485,7 @@ def train(hyp, opt, device, data_list,id,callbacks): # hyp is path/to/hyp.yaml # break # must break all DDP ranks ######实时传输训练精度参数############# #gain_train_report(int(epoch + 1), float(results[0]), pro, version, epochs) - report_cellback(epoch, epochs, float(results[0])) + # report_cellback(epoch, epochs, float(results[0])) # end epoch ---------------------------------------------------------------------------------------------------- ########训练数量和保存模型########### #save_train_report_result(pro, train_num, best)