manicman1999 / StyleGAN2-Tensorflow-2.0

StyleGAN 2 in Tensorflow 2.0
MIT License
486 stars 112 forks source link

Choice of loss function #11

Open 0xtristan opened 4 years ago

0xtristan commented 4 years ago

In the styleGAN 2 paper they state using a "non-saturating logistic loss", is there any particular reason you opted for hinge loss in this implementation?

For reference I believe the original loss functions for config-f are G_logistic_ns_pathreg for the generator and D_logistic_r1 for the discriminator (https://github.com/NVlabs/stylegan2/blob/master/training/loss.py).