diff --git a/train.py b/train.py
index 4039e29f..65151cec 100644
--- a/train.py
+++ b/train.py
@@ -446,7 +446,7 @@ if __name__ == '__main__':
         mixed_precision = False
 
     # scale hyp['obj'] by img_size (evolved at 320)
-    # hyp['obj'] *= opt.img_size / 320.
+    hyp['obj'] *= opt.img_size / 320.
 
     tb_writer = None
     if not opt.evolve:  # Train normally