From 5c067dc44785a761a0243d8cd634e3ac17d548ad Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Mon, 12 Sep 2016 20:55:20 +0000
Subject: [PATCH] good chance I didn't break anything

---
 cfg/alexnet.cfg |   58 ++++++++++++++++++++++++++++++++--------------------------
 1 files changed, 32 insertions(+), 26 deletions(-)

diff --git a/cfg/alexnet.cfg b/cfg/alexnet.cfg
index 8205c40..7e5a9b2 100644
--- a/cfg/alexnet.cfg
+++ b/cfg/alexnet.cfg
@@ -1,85 +1,91 @@
 [net]
 batch=128
 subdivisions=1
-height=256
-width=256
+height=227
+width=227
 channels=3
-learning_rate=0.01
 momentum=0.9
 decay=0.0005
+max_crop=256
 
-[crop]
-crop_height=224
-crop_width=224
-flip=1
-angle=0
-saturation=1
-exposure=1
+learning_rate=0.01
+policy=poly
+power=4
+max_batches=800000
+
+angle=7
+hue = .1
+saturation=.75
+exposure=.75
+aspect=.75
 
 [convolutional]
-filters=64
+filters=96
 size=11
 stride=4
 pad=0
-activation=ramp
+activation=relu
 
 [maxpool]
 size=3
 stride=2
+padding=0
 
 [convolutional]
-filters=192
+filters=256
 size=5
 stride=1
 pad=1
-activation=ramp
+activation=relu
 
 [maxpool]
 size=3
 stride=2
+padding=0
 
 [convolutional]
 filters=384
 size=3
 stride=1
 pad=1
-activation=ramp
+activation=relu
+
+[convolutional]
+filters=384
+size=3
+stride=1
+pad=1
+activation=relu
 
 [convolutional]
 filters=256
 size=3
 stride=1
 pad=1
-activation=ramp
-
-[convolutional]
-filters=256
-size=3
-stride=1
-pad=1
-activation=ramp
+activation=relu
 
 [maxpool]
 size=3
 stride=2
+padding=0
 
 [connected]
 output=4096
-activation=ramp
+activation=relu
 
 [dropout]
 probability=.5
 
 [connected]
 output=4096
-activation=ramp
+activation=relu
 
 [dropout]
 probability=.5
 
 [connected]
 output=1000
-activation=ramp
+activation=linear
 
 [softmax]
 groups=1

--
Gitblit v1.10.0