End-To-End Memory Network
- Python 2.7
- Torch (with nngraph)
First, preprocess included data into hdf5 format:
This will create a hdf5 file for each task (total 20 tasks).
This will train on task 16 with Linear Start over 100 epochs by default. See
train.lua (or the paper) for hyperparameters and more training options. To train on gpu, use