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

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

diff --git a/train.py b/train.py
index 0362a7a9..28636e53 100644
--- a/train.py
+++ b/train.py
@@ -127,6 +127,11 @@ 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:
+            x['lr'] = 0.1
+            x['momentum'] = 0.9
+
         for p in model.parameters():
             p.requires_grad = True if p.shape[0] == nf else False
 
@@ -364,6 +369,7 @@ if __name__ == '__main__':
     parser.add_argument('--weights', type=str, default='', help='initial weights')  # i.e. weights/darknet.53.conv.74
     opt = parser.parse_args()
     opt.weights = 'weights/last.pt' if opt.resume else opt.weights
+    opt.transfer = True
     print(opt)
 
     tb_writer = None