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]