/PyTorch-Dynamic_LSTM

implementation of AAAI 2019 Dynamic LSTM

Primary LanguageJupyter NotebookMIT LicenseMIT

PyTorch-Dynamic_LSTM

PyTorch implementation of Long Short-Term Memory with Dynamic Skip Connections (AAAI 2019).

Tensorflow implementation at https://github.com/lecholin/DynamicLSTM

Requirements

Python 3.6.5 PyTorch 1.0.0

Task

Only implemented the number prediction task. Currently working on language modeling task.

Citation

Please cite the original paper:

@article{gui2018long,
  title={Long Short-Term Memory with Dynamic Skip Connections},
  author={Gui, Tao and Zhang, Qi and Zhao, Lujun and Lin, Yaosong and Peng, Minlong and Gong, Jingjing and Huang, Xuanjing},
  journal={arXiv preprint arXiv:1811.03873},
  year={2018}
}