From 06e274f7e40b2cbf1827d4a57ceb8c5bc5804251 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Fri, 23 Aug 2019 16:04:45 +0200
Subject: [PATCH] updates

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

diff --git a/train.py b/train.py
index bb6367d9..ff8cfdf2 100644
--- a/train.py
+++ b/train.py
@@ -128,7 +128,7 @@ def train():
     if opt.transfer:  # transfer learning
         nf = int(model.module_defs[model.yolo_layers[0] - 1]['filters'])  # yolo layer size (i.e. 255)
 
-        for x in optimizer.param_groups:
+        for x in optimizer.param_groups:  # lower parameter count can handle more aggressive training hyps
             x['lr'] *= 10
             x['momentum'] *= 0.9