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
72 stars 18 forks source link

Using pretrained weights for the painter #169

Closed melisandeteng closed 3 years ago

melisandeteng commented 3 years ago

We use the weights from here to initialize our painter generator. The pretrained weights are from a SPADE model that does not have 7 upsampling layers as in the original paper but only 4, and the latent dim is 1024, and they have 182 classes. Thus we can only match the weights of the conv layers but not the weights for mlp beta and gamma parameters. We try several approaches: