From 7f8318e680f0308b4644762b7d9de66ec7c96aaf Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 23 Aug 2019 15:46:12 +0200 Subject: [PATCH] updates --- train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/train.py b/train.py index 28636e53..bb6367d9 100644 --- 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