Fixed densenet201_yolo.cfg - burn_in and poly policy, that changes learning rate

7 years ago · 4373f897f1
parent 8b385e1afe
commit 4373f897f1
1 changed files with 5 additions and 5 deletions
--- a/build/darknet/x64/densenet201_yolo.cfg
+++ b/build/darknet/x64/densenet201_yolo.cfg
@ -14,11 +14,11 @@ channels=3
 momentum=0.9
 decay=0.0005

-learning_rate=0.0001
-max_batches = 45000
-policy=steps
-steps=100,25000,35000
-scales=10,.1,.1
+burn_in=1000
+learning_rate=0.1
+policy=poly
+power=4
+max_batches=1600000

 angle=7
 hue=.1