Ankle/yolov5_ZWgra
1
0 Fork 0
Code Issues Pull Requests Actions 6 Packages Projects Releases Wiki Activity

Fix incorrect end epoch comment (#3612)

Browse Source
This commit is contained in:
Wei Quan authored and GitHub committed 2021-06-15 11:24:56 +02:00
1 parent daab682b06
commit 4c5d9bff80
1 file changed
-1
Download Patch File Download Diff File Expand all files Collapse all files
train.py
-1
View File
@@ -341,7 +341,6 @@ def train(hyp, opt, device, tb_writer=None):
save_dir.glob('train*.jpg') if x.exists()]}) save_dir.glob('train*.jpg') if x.exists()]})
# end batch ------------------------------------------------------------------------------------------------ # end batch ------------------------------------------------------------------------------------------------
# end epoch ----------------------------------------------------------------------------------------------------
# Scheduler # Scheduler
lr = [x['lr'] for x in optimizer.param_groups] # for tensorboard lr = [x['lr'] for x in optimizer.param_groups] # for tensorboard