From 579fdc57f89e1760e519e0d45e13195bd302e551 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Sat, 9 Nov 2019 10:56:38 -0800
Subject: [PATCH] updates

---
 train.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/train.py b/train.py
index 34658fd3..a38b9775 100644
--- a/train.py
+++ b/train.py
@@ -60,9 +60,6 @@ def train():
         hyp['cls_pw'] = 1.
         hyp['obj_pw'] = 1.
 
-    if img_size > 320:  # scale hyp['obj'] by img_size (evolved at 320)
-        hyp['obj'] *= img_size / 320.
-
     # Initialize
     init_seeds()
     multi_scale = opt.multi_scale
@@ -422,6 +419,9 @@ if __name__ == '__main__':
     print(opt)
     device = torch_utils.select_device(opt.device, apex=mixed_precision)
 
+    # scale hyp['obj'] by img_size (evolved at 320)
+    hyp['obj'] *= opt.img_size / 320.
+
     tb_writer = None
     if not opt.evolve:  # Train normally
         try: