search

cc-ai / climategan

Code and pre-trained model for the algorithm generating visualisations of 3 climate change related events: floods, wildfires and smog.
https://thisclimatedoesnotexist.com
GNU General Public License v3.0
75 stars 18 forks source link

total variational training loss #62

Closed 51N84D closed 4 years ago

51N84D commented 4 years ago

Adding total variational regularization for smoothing mask predictions (in the mask decoder)

51N84D commented 4 years ago

I'm trying to allow multiple loss functions per update task. I'm sure there's a cleaner way to do it than this. What do you guys think?

vict0rsch commented 4 years ago

I'm trying to allow multiple loss functions per update task. I'm sure there's a cleaner way to do it than this. What do you guys think?

can you elaborate? I'm not sure I see the issue