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 | 28 ++++++++++++++++++++-------- 1 files changed, 20 insertions(+), 8 deletions(-) diff --git a/cfg/darknet.cfg b/cfg/darknet.cfg index 2e0a624..53d1ec9 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 @@ -17,6 +21,7 @@ exposure=1 [convolutional] +batch_normalize=1 filters=16 size=3 stride=1 @@ -24,10 +29,11 @@ activation=leaky [maxpool] -stride=2 size=2 +stride=2 [convolutional] +batch_normalize=1 filters=32 size=3 stride=1 @@ -35,10 +41,11 @@ activation=leaky [maxpool] -stride=2 size=2 +stride=2 [convolutional] +batch_normalize=1 filters=64 size=3 stride=1 @@ -46,10 +53,11 @@ activation=leaky [maxpool] -stride=2 size=2 +stride=2 [convolutional] +batch_normalize=1 filters=128 size=3 stride=1 @@ -57,10 +65,11 @@ activation=leaky [maxpool] -stride=2 size=2 +stride=2 [convolutional] +batch_normalize=1 filters=256 size=3 stride=1 @@ -68,10 +77,11 @@ activation=leaky [maxpool] -stride=2 size=2 +stride=2 [convolutional] +batch_normalize=1 filters=512 size=3 stride=1 @@ -79,10 +89,11 @@ activation=leaky [maxpool] -stride=2 size=2 +stride=2 [convolutional] +batch_normalize=1 filters=1024 size=3 stride=1 @@ -99,6 +110,7 @@ activation=leaky [softmax] +groups=1 [cost] type=sse -- Gitblit v1.10.0