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

Switch set_losses() to losses.py #65

Closed alexrey88 closed 4 years ago

alexrey88 commented 4 years ago

Small branch to remove set_losses() from Trainer class and add it as a pure function in losses.py.