updates

2019-08-09 18:22:27 +02:00 · 2019-08-09 18:22:27 +02:00 · 298463e530
commit 298463e530
parent 0c52cc0106
1 changed files with 2 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -193,7 +193,8 @@ def train(cfg,
              ('Epoch', 'gpu_mem', 'GIoU/xy', 'wh', 'obj', 'cls', 'total', 'targets', 'img_size'))

        # Update scheduler
-        scheduler.step()
+        if epoch > 0:
+            scheduler.step()

        # Freeze backbone at epoch 0, unfreeze at epoch 1 (optional)
        freeze_backbone = False