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

TODO setup a get_losses func to assign the right loss according to the yaml #63

Closed vict0rsch closed 4 years ago

vict0rsch commented 4 years ago

def set_losses(self): doesn't really ned to be a trainer method, it could be a pure function in tutils that takes opts and returns a losses dict which we would set to be a trainer attribute somewhere in its __init__()

alexrey88 commented 4 years ago

I could do this, if that's ok with you. :) But why tutils and not utils tho, it's not tensor-related, is it?

vict0rsch commented 4 years ago

great! :)

I mean it's not tensors but it's pytorch. It's a moving target right, it doesn't matter so much, worst case scenario we'll just move it around :p

alexrey88 commented 4 years ago

true true!