/ECE176_Project

ECE 176 course project

Primary LanguageJupyter NotebookApache License 2.0Apache-2.0

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