Added alexnet cfg

This commit is contained in:
Joseph Redmon 2015-06-25 10:26:47 -07:00
parent 5b0166bf88
commit d1d56a2a72

90
cfg/alexnet.cfg Normal file
View File

@ -0,0 +1,90 @@
[net]
batch=128
subdivisions=1
height=256
width=256
channels=3
learning_rate=0.01
momentum=0.9
decay=0.0005
seen=0
[crop]
crop_height=224
crop_width=224
flip=1
angle=0
saturation=1
exposure=1
[convolutional]
filters=64
size=11
stride=4
pad=0
activation=ramp
[maxpool]
size=3
stride=2
[convolutional]
filters=192
size=5
stride=1
pad=1
activation=ramp
[maxpool]
size=3
stride=2
[convolutional]
filters=384
size=3
stride=1
pad=1
activation=ramp
[convolutional]
filters=256
size=3
stride=1
pad=1
activation=ramp
[convolutional]
filters=256
size=3
stride=1
pad=1
activation=ramp
[maxpool]
size=3
stride=2
[connected]
output=4096
activation=ramp
[dropout]
probability=.5
[connected]
output=4096
activation=ramp
[dropout]
probability=.5
[connected]
output=1000
activation=ramp
[softmax]
groups=1
[cost]
type=sse