From c2738835f0a2435ab03f411af3d168aec389d2a6 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 08 Dec 2015 01:18:04 +0000
Subject: [PATCH] Faster batch normalization

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

diff --git a/cfg/darknet.cfg b/cfg/darknet.cfg
index 2e0a624..85497a3 100644
--- a/cfg/darknet.cfg
+++ b/cfg/darknet.cfg
@@ -1,13 +1,17 @@
 [net]
 batch=128
-subdivisions=32
+subdivisions=1
 height=256
 width=256
 channels=3
-learning_rate=0.01
 momentum=0.9
 decay=0.0005
 
+learning_rate=0.1
+policy=poly
+power=4
+max_batches=500000
+
 [crop]
 crop_height=224
 crop_width=224
@@ -24,8 +28,8 @@
 activation=leaky
 
 [maxpool]
-stride=2
 size=2
+stride=2
 
 [convolutional]
 filters=32
@@ -35,8 +39,8 @@
 activation=leaky
 
 [maxpool]
-stride=2
 size=2
+stride=2
 
 [convolutional]
 filters=64
@@ -46,8 +50,8 @@
 activation=leaky
 
 [maxpool]
-stride=2
 size=2
+stride=2
 
 [convolutional]
 filters=128
@@ -57,8 +61,8 @@
 activation=leaky
 
 [maxpool]
-stride=2
 size=2
+stride=2
 
 [convolutional]
 filters=256
@@ -68,8 +72,8 @@
 activation=leaky
 
 [maxpool]
-stride=2
 size=2
+stride=2
 
 [convolutional]
 filters=512
@@ -79,8 +83,8 @@
 activation=leaky
 
 [maxpool]
-stride=2
 size=2
+stride=2
 
 [convolutional]
 filters=1024
@@ -99,6 +103,7 @@
 activation=leaky
 
 [softmax]
+groups=1
 
 [cost]
 type=sse

--
Gitblit v1.10.0