liaohaofu / adn

ADN: Artifact Disentanglement Network for Unsupervised Metal Artifact Reduction
Other
164 stars 39 forks source link

THE WGANGP is not right? #7

Closed brisyramshere closed 4 years ago

brisyramshere commented 4 years ago

have no gradient penalty term?

liaohaofu commented 4 years ago

Hi, the GANLoss class is adopted from the CycleGAN repo which provides WGANGP. In our work, we use LSGAN by default and WGANGP is not fully supported. You may refer here for the GP calculation.

brisyramshere commented 4 years ago

thank you very much