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 | 68 ++++++---------------------------- 1 files changed, 12 insertions(+), 56 deletions(-) diff --git a/cfg/darknet.cfg b/cfg/darknet.cfg index 8a5978e..85497a3 100644 --- a/cfg/darknet.cfg +++ b/cfg/darknet.cfg @@ -4,10 +4,14 @@ 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 @@ -23,16 +27,9 @@ pad=1 activation=leaky -#[convolutional] -#filters=16 -#size=3 -#stride=1 -#pad=1 -#activation=leaky - [maxpool] -stride=2 size=2 +stride=2 [convolutional] filters=32 @@ -41,16 +38,9 @@ pad=1 activation=leaky -#[convolutional] -#filters=32 -#size=3 -#stride=1 -#pad=1 -#activation=leaky - [maxpool] -stride=2 size=2 +stride=2 [convolutional] filters=64 @@ -59,16 +49,9 @@ pad=1 activation=leaky -#[convolutional] -#filters=64 -#size=3 -#stride=1 -#pad=1 -#activation=leaky - [maxpool] -stride=2 size=2 +stride=2 [convolutional] filters=128 @@ -77,16 +60,9 @@ pad=1 activation=leaky -#[convolutional] -#filters=128 -#size=3 -#stride=1 -#pad=1 -#activation=leaky - [maxpool] -stride=2 size=2 +stride=2 [convolutional] filters=256 @@ -95,16 +71,9 @@ pad=1 activation=leaky -#[convolutional] -#filters=256 -#size=3 -#stride=1 -#pad=1 -#activation=leaky - [maxpool] -stride=2 size=2 +stride=2 [convolutional] filters=512 @@ -113,16 +82,9 @@ pad=1 activation=leaky -#[convolutional] -#filters=512 -#size=3 -#stride=1 -#pad=1 -#activation=leaky - [maxpool] -stride=2 size=2 +stride=2 [convolutional] filters=1024 @@ -131,13 +93,6 @@ pad=1 activation=leaky -#[convolutional] -#filters=1024 -#size=3 -#stride=1 -#pad=1 -#activation=leaky - [avgpool] [dropout] @@ -148,6 +103,7 @@ activation=leaky [softmax] +groups=1 [cost] type=sse -- Gitblit v1.10.0