Joseph Redmon
2016-09-12 5c067dc44785a761a0243d8cd634e3ac17d548ad
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]