SNGAN-AdaBelief

KO | EN

gitlite — search

#python #java #python3 #arduino #golang #machine-learning #rust #html #flask #javascript #seismology #nodejs

SNGAN-AdaBelief

★ 11 Open GitHub ↗

Train SN-GAN with AdaBelief

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

## 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     |

![](https://raw.githubusercontent.com/juntang-zhuang/SNGAN-AdaBelief/master/logs/fid.png)

SNGAN-AdaBelief

Explore Similar Repositories

Was this content helpful?

Recent Feedback

Download README