From 1ea87c49c4e7ff368dfd31a7db4c4374b9ae7507 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Thu, 15 Nov 2018 00:57:15 +0100
Subject: [PATCH] updates

---
 train.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/train.py b/train.py
index dfe30c8f..59d00b19 100644
--- a/train.py
+++ b/train.py
@@ -105,9 +105,9 @@ def main(opt):
 
         # Update scheduler (manual)  at 0, 54, 61 epochs to 1e-3, 1e-4, 1e-5
         if epoch > 50:
-            lr = 1e-5
-        else:
             lr = 1e-4
+        else:
+            lr = 1e-3
         for g in optimizer.param_groups:
             g['lr'] = lr