darknet/cfg/rnn.train.cfg
Joseph Redmon c7c1e0e7b7 rnn stuff
2016-02-05 00:15:12 -08:00

41 lines
437 B
INI

[net]
subdivisions=1
inputs=256
batch = 128
momentum=0.9
decay=0.001
max_batches = 2000
time_steps=576
learning_rate=0.1
policy=steps
steps=1000,1500
scales=.1,.1
[rnn]
batch_normalize=1
output = 1024
hidden=1024
activation=leaky
[rnn]
batch_normalize=1
output = 1024
hidden=1024
activation=leaky
[rnn]
batch_normalize=1
output = 1024
hidden=1024
activation=leaky
[connected]
output=256
activation=leaky
[softmax]
[cost]
type=sse