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

[WIP] Include simCLR #80

Closed alexrey88 closed 4 years ago

alexrey88 commented 4 years ago

The idea is to implement simCLR in the code: train the base encoder with simCLR, and then fine-tune the weights of the encoder with our main task (the masker). See if we get better results than with current deeplabv2 encoder.

alexrey88 commented 4 years ago

Reminder : add art option for pretraining, and move simclr loss computation to another function get_pretraining_loss