updates

2019-08-23 15:46:12 +02:00 · 2019-08-23 15:46:12 +02:00 · 7f8318e680
parent f0622e2510
commit 7f8318e680
1 changed files with 2 additions and 2 deletions
--- a/train.py
+++ b/train.py
@ -129,8 +129,8 @@ def train():
        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
+            x['lr'] *= 10
+            x['momentum'] *= 0.9

        for p in model.parameters():
            p.requires_grad = True if p.shape[0] == nf else False