cfg/rnn.cfg
@@ -1,29 +1,32 @@ [net] subdivisions=1 inputs=256 batch = 128 batch = 1 momentum=0.9 decay=0.001 max_batches = 50000 time_steps=900 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]
