README.md

Requirements

pytorch >= 0.4.1

Setup Environment

# Dependencies
pip install matplotlib
pip install git+https://github.com/wookayin/tensorflow-plot.git@master

Dataset

Download small NORB dataset

cd scripts
./download_norb_small.sh

Train

./run.sh

Results

Acknowledgment

https://github.com/ndrplz/small_norb