From e7d43fd65ddc476469ee8d24140835c1e0159fa6 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Mon, 30 Nov 2015 23:04:09 +0000
Subject: [PATCH] rolling avg demo

---
 cfg/darknet.cfg |   19 +++++++++++++------
 1 files changed, 13 insertions(+), 6 deletions(-)

diff --git a/cfg/darknet.cfg b/cfg/darknet.cfg
index 64aab1e..53d1ec9 100644
--- a/cfg/darknet.cfg
+++ b/cfg/darknet.cfg
@@ -1,5 +1,5 @@
 [net]
-batch=256
+batch=128
 subdivisions=1
 height=256
 width=256
@@ -7,11 +7,10 @@
 momentum=0.9
 decay=0.0005
 
-learning_rate=0.01
-policy=step
-scale=.1
-step=100000
-max_batches=400000
+learning_rate=0.1
+policy=poly
+power=4
+max_batches=500000
 
 [crop]
 crop_height=224
@@ -22,6 +21,7 @@
 exposure=1
 
 [convolutional]
+batch_normalize=1
 filters=16
 size=3
 stride=1
@@ -33,6 +33,7 @@
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=32
 size=3
 stride=1
@@ -44,6 +45,7 @@
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=64
 size=3
 stride=1
@@ -55,6 +57,7 @@
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=128
 size=3
 stride=1
@@ -66,6 +69,7 @@
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=256
 size=3
 stride=1
@@ -77,6 +81,7 @@
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=3
 stride=1
@@ -88,6 +93,7 @@
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=1024
 size=3
 stride=1
@@ -104,6 +110,7 @@
 activation=leaky
 
 [softmax]
+groups=1
 
 [cost]
 type=sse

--
Gitblit v1.10.0