From ea61b46b3173359164b29e983c7012059c9a5054 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 2 Sep 2019 14:02:08 +0200 Subject: [PATCH] updates Signed-off-by: Glenn Jocher --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 21b8e23b..996e48da 100644 --- a/train.py +++ b/train.py @@ -80,7 +80,7 @@ def train(): # optimizer = optim.Adam(pg0, lr=hyp['lr0']) # optimizer = AdaBound(pg0, lr=hyp['lr0'], final_lr=0.1) - # optimizer = optim.SGD(pg0, lr=hyp['lr0'], momentum=hyp['momentum'], nesterov=True) + optimizer = optim.SGD(pg0, lr=hyp['lr0'], momentum=hyp['momentum'], nesterov=True) optimizer.add_param_group({'params': pg1, 'weight_decay': hyp['weight_decay']}) # add pg1 with weight_decay del pg0, pg1