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

infer.py not working #135

Closed melisandeteng closed 3 years ago

melisandeteng commented 3 years ago

We need to check what is going on with infer.py :

RuntimeError: Error(s) in loading state_dict for OmniGenerator:
    Missing key(s) in state_dict: ... 
melisandeteng commented 3 years ago

Actually this was due to mismatching yaml file from before the issue of saving yaml files associated to checkpoints was fixed