From ef3bd7e12b27b653d89a7fd0fed9913c8c9e3ea0 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Mon, 24 Feb 2020 09:06:17 -0800
Subject: [PATCH] updates

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

diff --git a/train.py b/train.py
index 0c63a514..24a2f7ab 100644
--- a/train.py
+++ b/train.py
@@ -140,7 +140,7 @@ def train():
     # scheduler = lr_scheduler.LambdaLR(optimizer, lr_lambda=lf)
     # scheduler = lr_scheduler.MultiStepLR(optimizer, milestones=range(59, 70, 1), gamma=0.8)  # gradual fall to 0.1*lr0
     scheduler = lr_scheduler.MultiStepLR(optimizer, milestones=[round(epochs * x) for x in [0.8, 0.9]], gamma=0.1)
-    scheduler.last_epoch = start_epoch - 1
+    scheduler.last_epoch = start_epoch
 
     # # Plot lr schedule
     # y = []