SNGAN-AdaBelief
Train a state-of-the-art spectral normalization GAN with AdaBelief https://github.com/juntang-zhuang/Adabelief-Optimizer
Acknowledgement
This repo is forked from PyTorch-StudioGAN
github repository, with the only difference in optimizer.
Dependencies
pip install adabelief-pytorch==0.0.5
For other dependencies, see https://github.com/POSTECH-CVLab/PyTorch-StudioGAN/blob/master/environment.yml
How to run
python main.py -t -e -c configs/CIFAR10/SNGAN-adabelief.json
You can modify parameters epsilon
and rectify
in load_framwork.py
https://github.com/juntang-zhuang/SNGAN-AdaBelief/blob/master/load_framework.py#L135
Results
Results for Adam is directly taken from the training log of official implementation https://github.com/POSTECH-CVLab/PyTorch-StudioGAN/blob/c98d0e94d98a97e14165bd42bd1416027cf78f4d/logs/CIFAR10/SNGAN-train-2020_09_18_14_37_00.log
Adam | AdaBelief (eps = 1e-12, rectify=False) | |
---|---|---|
FID | 13.25 | 12.87 |