From 298463e530ec16a654d6f3b32279941ed0fa4d75 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Fri, 9 Aug 2019 18:22:27 +0200
Subject: [PATCH] updates

---
 train.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/train.py b/train.py
index aa813d83..b13e8e75 100644
--- 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