From c91ffee8524bbc639aa343bfc29856a806d5174b Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 19 Jan 2020 16:55:29 -0800 Subject: [PATCH] updates --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index ba1ed5a0..e9987474 100644 --- a/train.py +++ b/train.py @@ -94,7 +94,7 @@ def train(): pg0 += [v] # all else if opt.adam: - hyp['lr0'] *= 0.1 # reduce lr (i.e. SGD=5E-3, Adam=5E-4) + # hyp['lr0'] *= 0.1 # reduce lr (i.e. SGD=5E-3, Adam=5E-4) optimizer = optim.Adam(pg0, lr=hyp['lr0']) # optimizer = AdaBound(pg0, lr=hyp['lr0'], final_lr=0.1) else: