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