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 | 22 ++++++++++++---------- 1 files changed, 12 insertions(+), 10 deletions(-) diff --git a/cfg/darknet.cfg b/cfg/darknet.cfg index f52ff3f..85497a3 100644 --- a/cfg/darknet.cfg +++ b/cfg/darknet.cfg @@ -6,10 +6,11 @@ channels=3 momentum=0.9 decay=0.0005 -learning_rate=0.01 + +learning_rate=0.1 policy=poly -power=.5 -max_batches=600000 +power=4 +max_batches=500000 [crop] crop_height=224 @@ -27,7 +28,7 @@ activation=leaky [maxpool] -size=3 +size=2 stride=2 [convolutional] @@ -38,7 +39,7 @@ activation=leaky [maxpool] -size=3 +size=2 stride=2 [convolutional] @@ -49,7 +50,7 @@ activation=leaky [maxpool] -size=3 +size=2 stride=2 [convolutional] @@ -60,7 +61,7 @@ activation=leaky [maxpool] -size=3 +size=2 stride=2 [convolutional] @@ -71,7 +72,7 @@ activation=leaky [maxpool] -size=3 +size=2 stride=2 [convolutional] @@ -82,7 +83,7 @@ activation=leaky [maxpool] -size=3 +size=2 stride=2 [convolutional] @@ -99,9 +100,10 @@ [connected] output=1000 -activation=linear +activation=leaky [softmax] +groups=1 [cost] type=sse -- Gitblit v1.10.0