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: