/NIPS16-CGMMN

Part codes for CGMMN

Primary LanguagePython

  • to classify MNIST dataset with CNN layers just run "./run-cnn.sh"

  • to generate MNIST samples, you should firstly set "pre-train = 0" in "cmmd-gen.py" and run "./run-gen.sh"; then set "pre-train =1", load the trained model and run "./run-gen/sh" again.