- The code is widely adapted from two nice repositories:
https://github.com/Abhipanda4/GMMN-Pytorch
https://github.com/siddharth-agrawal/Generative-Moment-Matching-Networks. - I failed to implement the greedy layer-wise pretraining and fine-tuning scheme mentioned in the paper as I have not fully understood it.
- CentOS Linux release 7.2.1511 (Core)
- python 3.6.5
- pytorch 1.0.0
- torchvision
- argparse
- pickle
python train.py
Two folders will be created, i.e., ./data
& ./models
. As their names imply, they are used to store data and trained models, repectively.
python Visualize.py --visualize gmmn
python Visualize.py --visualize autoencoder
https://github.com/Abhipanda4/GMMN-Pytorch
https://github.com/siddharth-agrawal/Generative-Moment-Matching-Networks