astoken commited on
Commit
bc4ef48
·
1 Parent(s): 7abf202

Default optimizer SGD

Browse files
Files changed (1) hide show
  1. train.py +1 -1
train.py CHANGED
@@ -20,7 +20,7 @@ except:
20
 
21
 
22
  # Hyperparameters
23
- hyp = {'optimizer': 'adam' #if none, default is SGD
24
  'lr0': 0.01, # initial learning rate (SGD=1E-2, Adam=1E-3)
25
  'momentum': 0.937, # SGD momentum/Adam beta1
26
  'weight_decay': 5e-4, # optimizer weight decay
 
20
 
21
 
22
  # Hyperparameters
23
+ hyp = {'optimizer': 'SGD', # ['adam, 'SGD', None] if none, default is SGD
24
  'lr0': 0.01, # initial learning rate (SGD=1E-2, Adam=1E-3)
25
  'momentum': 0.937, # SGD momentum/Adam beta1
26
  'weight_decay': 5e-4, # optimizer weight decay