Default optimizer SGD

5 years ago · bc4ef4861b
parent 7abf202cad
commit bc4ef4861b
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -20,7 +20,7 @@ except:
 # Hyperparameters
-hyp = {'optimizer': 'adam' #if none, default is SGD
+hyp = {'optimizer': 'SGD', # ['adam, 'SGD', None] if none, default is SGD
       'lr0': 0.01,  # initial learning rate (SGD=1E-2, Adam=1E-3)
       'momentum': 0.937,  # SGD momentum/Adam beta1
       'weight_decay': 5e-4,  # optimizer weight decay