Changes:
- adding noise to images
- label smoothing
Things to try:
- spectral norm
- adaptive residuals
pip3 install -r requirements.txt
python3 train.py
Then you can compute LPIPS by running:
python3 test.py --checkpoint checkpoints/my/checkpoint.pt
Open visualizer.ipynb to visualize results.
Undertrained (10k steps) checkpoint can be downloaded via:
gdown --fuzzy https://drive.google.com/file/d/1CJwZOoVV8aCa2Jvn2cBoDpJLZXnI3KjN/view?usp=sharing