- Linux or macOS
- Python 3
- CPU or NVIDIA GPU + CUDA CuDNN
- Preparing .npz files for Pytorch Inception metrics evaluation (cifar10 as an example):
python envs_repo/inception_pytorch/calculate_inception_moments.py --dataset C10 --data_root datasets
- Preparing .tgz files for Tensorflow Inception metrics evaluation
- Preparing .npz files for Tensorflow FID metrics evaluation
An example of GAN training command was saved in ./scripts/operator_cifar10.sh. Train a model (cifar10 as an example):
bash ./scripts/operator_cifar10.sh
An example of CE-GAN training command was saved in ./scripts/cegan_cifar10.sh. Train a model (cifar10 as an example):
bash ./scripts/cegan_cifar10.sh
Pytorch Inception metrics code from BigGAN-PyTorch.
TensorFlow Inception Score code from OpenAI's Improved-GAN..
TensorFlow FID code from TTUR.