RNN Example

This folder contains RNN examples using low level symbol interface.

Data

Run get_ptb_data.sh to download PenTreeBank data.

Python

• lstm.py Functions for building a LSTM Network
• gru.py Functions for building a GRU Network
• lstm_bucketing.py PennTreeBank language model by using LSTM
• gru_bucketing.py PennTreeBank language model by using GRU
• char-rnn.ipynb Notebook to demo how to train a character LSTM by using lstm.py

Performance Note: More MXNET_GPU_WORKER_NTHREADS may lead to better performance. For setting MXNET_GPU_WORKER_NTHREADS, please refer to Environment Variables.