1
0

rnn.train.cfg 421 B

1234567891011121314151617181920212223242526272829303132333435363738
  1. [net]
  2. subdivisions=1
  3. inputs=256
  4. batch = 128
  5. momentum=0.9
  6. decay=0.001
  7. max_batches = 2000
  8. time_steps=576
  9. learning_rate=0.1
  10. policy=steps
  11. steps=1000,1500
  12. scales=.1,.1
  13. [rnn]
  14. batch_normalize=1
  15. output = 1024
  16. hidden=1024
  17. activation=leaky
  18. [rnn]
  19. batch_normalize=1
  20. output = 1024
  21. hidden=1024
  22. activation=leaky
  23. [rnn]
  24. batch_normalize=1
  25. output = 1024
  26. hidden=1024
  27. activation=leaky
  28. [connected]
  29. output=256
  30. activation=leaky
  31. [softmax]