gru.cfg 324 B

12345678910111213141516171819202122232425262728293031323334
  1. [net]
  2. subdivisions=1
  3. inputs=256
  4. batch = 1
  5. momentum=0.9
  6. decay=0.001
  7. time_steps=1
  8. learning_rate=0.5
  9. policy=poly
  10. power=4
  11. max_batches=2000
  12. [gru]
  13. batch_normalize=1
  14. output = 1024
  15. [gru]
  16. batch_normalize=1
  17. output = 1024
  18. [gru]
  19. batch_normalize=1
  20. output = 1024
  21. [connected]
  22. output=256
  23. activation=linear
  24. [softmax]
  25. [cost]
  26. type=sse