train_classifier.py
trains an oracle classifier on the MNIST dataset -> This is what we will use for evaluation.train_cgan.ipynb
trains a standard cGAN on MNIST.mnist_cnn.pth
is the oracle classifier.
- Pretrain classifier with CPC, and then just slightly tune it on the noisy target labels.
- Setup normal cGAN training loop for pretraining the discriminator and generator.
- Switch to finetuning the classifier with the generator and discriminator.