Skip to content

Latest commit

 

History

History

rnn

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 
 
 

Train RNN model over IMDB dataset

Recurrent neural networks (RNN) are widely used for modelling sequential data, e.g., natural language sentences. This example describes how to implement a RNN application (or model) using SINGA's CUDNN RNN layers. We will use the LSTM model as an example to train on IMDB dataset.

Instructions

  • Prepare the dataset,

      python imdb_data.py
    
  • Start the training,

      python imdb_train.py