From e5b5d6a880fcd354e515c8d3b6d7eb2861b4422d Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 27 Dec 2019 14:58:31 -0800 Subject: [PATCH] updates --- train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/train.py b/train.py index 7a9d2d07..65151cec 100644 --- a/train.py +++ b/train.py @@ -142,9 +142,9 @@ def train(): if opt.prebias: for p in optimizer.param_groups: # lower param count allows more aggressive training settings: i.e. SGD ~0.1 lr0, ~0.9 momentum - p['lr'] *= 100 # lr gain + p['lr'] = 0.1 # learning rate if p.get('momentum') is not None: # for SGD but not Adam - p['momentum'] *= 0.9 + p['momentum'] = 0.9 for p in model.parameters(): if opt.prebias and p.numel() == nf: # train (yolo biases)