search

AAnoosheh / ToDayGAN

http://arxiv.org/abs/1809.09767
BSD 2-Clause "Simplified" License
172 stars 32 forks source link

About the discriminator #4

Closed ArcherrrrR closed 5 years ago

ArcherrrrR commented 5 years ago

Hello, I have just read your paper and try to run your code. For the three discriminators, you mention in the paper that the loss are averaged in the end equally And I want to know what should I change in the code if I want to put more weight on the gradient discriminator. For I am just a new hand in coding and don't know how to do it. Hope you can help me! Thanks a lot!

AAnoosheh commented 5 years ago

Hi,

Line 45 of combogan_model.py contains the averaging you are looking for. The order of the three losses in there are RGB, Gray, and Gradients, respectively, so you can change the weighting as you need.

Note that I have tried 0.25,0.25,0.5 in the past and got significantly worse results.