Default optimizer SGD
Browse files
train.py
CHANGED
@@ -20,7 +20,7 @@ except:
|
|
20 |
|
21 |
|
22 |
# Hyperparameters
|
23 |
-
hyp = {'optimizer': 'adam'
|
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
|