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

Segmentation head #106

Closed alexrey88 closed 4 years ago

alexrey88 commented 4 years ago

Adding segmentation head as an auxiliary task is the purpose of this PR. :)

alexrey88 commented 4 years ago

All right. Segmentation head is running well. The results are not as good as I thought they would be but maybe there's something I've missed in the code, or maybe I should play with the learning rate/batch_size more. Here's the current running experiment : here. Waiting for you reviews !! :)