Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

Default optimizer SGD

Browse Source
This commit is contained in:
Alex Stoken committed 2020-06-24 10:07:43 -05:00
1 parent 7abf202cad
commit bc4ef4861b
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
train.py
+1 -1
View File
@@ -20,7 +20,7 @@ except:
# Hyperparameters # 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) 'lr0': 0.01, # initial learning rate (SGD=1E-2, Adam=1E-3)
'momentum': 0.937, # SGD momentum/Adam beta1 'momentum': 0.937, # SGD momentum/Adam beta1
'weight_decay': 5e-4, # optimizer weight decay 'weight_decay': 5e-4, # optimizer weight decay