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