Models tested on MarioQA dataset

October 6, 2017 ยท View on GitHub

1. Dependencies (This project is tested on linux 14.04 64bit with gpu of Titan X and python 2.7.12 with anaconda 4.1.9)

Dependencies for torch (need for model training and evaluation)

Dependencies for python (need for data pre-processing)

1. Setup instruction

We need two following processes in 001_data_constuction.

Perform each process in proper folder by following the README file.

2. Training and evaluating models

Model training

Move to folder that you want to train the model, and run following lines.

mv 003_neural_models/003_spatio_temporal_attention/
bash gen_simulinks.sh
bash run_train.sh

Model evaluation

Move to folder that you want to evaluate the model, and run following lines.

mv 003_neural_models/003_spatio_temporal_attention/
bash gen_simulinks.sh
bash run_download_pretrained_model.sh
bash run_eval.sh