znxlwm / UGATIT-pytorch

Official PyTorch implementation of U-GAT-IT: Unsupervised Generative Attentional Networks with Adaptive Layer-Instance Normalization for Image-to-Image Translation
MIT License
2.51k stars 476 forks source link

CAM loss incorrect? #33

Closed FangYang970206 closed 5 years ago

FangYang970206 commented 5 years ago

Hi, in the paper, I find $L{cam}^{D{t}}$ have a log function, but in source code, only use MSE(UGATIT.py 182-189 lines). So, the paper adds unnecessarily log?

znxlwm commented 5 years ago

Hi,

The log is a typo in the paper.

We used MSE as CAM loss of discriminator.

Thanks.