rnn.cfg 433 B

12345678910111213141516171819202122232425262728293031323334353637383940
  1. [net]
  2. subdivisions=1
  3. inputs=256
  4. batch = 1
  5. momentum=0.9
  6. decay=0.001
  7. max_batches = 2000
  8. time_steps=1
  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]
  32. [cost]
  33. type=sse