Skip to content

Latest commit

 

History

History

rnn

Implement RNN (Vanilla, GRU, LSTM, LSTMP) from scratch.

  • modules.py - define RNN variants
  • test_rnn.py - demo

Gradient is clipped to avoid explosion using pytorch Variable's register_hook function.