Edmond Yoo
2018-09-16 176260d82a4d82ce4ce1f09cd6139a50e1a2aa84
cfg/alexnet.cfg
@@ -1,85 +1,91 @@
[net]
batch=128
subdivisions=128
height=256
width=256
subdivisions=1
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