Fix incorrect end epoch comment (#3612)

2021-06-15 05:24:56 -04:00 · 2021-06-15 05:24:56 -04:00 · 4c5d9bff80
parent daab682b06
commit 4c5d9bff80
1 changed files with 1 additions and 2 deletions
--- a/train.py
+++ b/train.py
@ -341,8 +341,7 @@ def train(hyp, opt, device, tb_writer=None):
                                                  save_dir.glob('train*.jpg') if x.exists()]})

            # end batch ------------------------------------------------------------------------------------------------
-        # end epoch ----------------------------------------------------------------------------------------------------
-
+        
        # Scheduler
        lr = [x['lr'] for x in optimizer.param_groups]  # for tensorboard
        scheduler.step()