This folder contains RNN examples using low level symbol interface.
get_ptb_data.sh to download PenTreeBank data.
- 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
MXNET_GPU_WORKER_NTHREADS may lead to better performance. For setting
MXNET_GPU_WORKER_NTHREADS, please refer to Environment Variables.