Joseph Redmon
2016-02-05 01f20abe847768a8fb3e27739a284b541eb08ff6
cfg/rnn.cfg
@@ -2,28 +2,31 @@
subdivisions=1
inputs=256
batch = 1
time_steps = 1
momentum=0.9
decay=0.0005
max_batches = 50000000
decay=0.001
max_batches = 2000
time_steps=1
learning_rate=0.1
policy=steps
steps=1000,1500
scales=.1,.1
[rnn]
batch_normalize=1
output = 256
hidden=512
output = 1024
hidden=1024
activation=leaky
[rnn]
batch_normalize=1
output = 256
hidden=512
output = 1024
hidden=1024
activation=leaky
[rnn]
batch_normalize=1
output = 256
hidden=512
output = 1024
hidden=1024
activation=leaky
[connected]