Latent Constraints for Data Generation

Requirements

• Python >= 3.5
• PyTorch >= 1.3
• Pytorch Lightning >= 0.6.0 (GitHub Repo)
• CUDA enabled computing device

Usage

$ python run.py -c configs/<config-file-name.yaml>

run.py - train vae model on CIFAR10
run_vaemnist.py - train vae model on MNIST
run_ac.py - train actor critic model
run_ac_mnist.py - train actor critic model for MNIST
train_classifier - train the classifier on CIFAR 10
train_classifier_mnist - train the classifier on MNIST