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

Fixing print(model) functions #53

Closed 51N84D closed 4 years ago

51N84D commented 4 years ago

Some versions of pytorch don't have the ".weight" attribute for Conv2D objects. This code just explicitly computes it to support model printing (omnigan.strings)