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

实时返回训练loss report

Browse Source
This commit is contained in:
JIAKUNHAO
2022-11-18 15:20:33 +08:00
parent 59cc951d63
commit 950e29eb49
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
app/yolov5/train_server.py
View File

@@ -400,6 +400,7 @@ def train(hyp, opt, device, data_list,id,callbacks): # hyp is path/to/hyp.yaml
# Backward # Backward
scaler.scale(loss).backward() scaler.scale(loss).backward()
report_cellback(epoch, epochs, float(loss))
# Optimize - https://pytorch.org/docs/master/notes/amp_examples.html # Optimize - https://pytorch.org/docs/master/notes/amp_examples.html
if ni - last_opt_step >= accumulate: 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 # break # must break all DDP ranks
######实时传输训练精度参数############# ######实时传输训练精度参数#############
#gain_train_report(int(epoch + 1), float(results[0]), pro, version, epochs) #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 ---------------------------------------------------------------------------------------------------- # end epoch ----------------------------------------------------------------------------------------------------
########训练数量和保存模型########### ########训练数量和保存模型###########
#save_train_report_result(pro, train_num, best) #save_train_report_result(pro, train_num, best)