Add Classifier - Githubissues

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

76 stars 18 forks source link

Closed vict0rsch closed 4 years ago

vict0rsch commented 4 years ago

Create real classifier architecture in classifier.py
Add / adapt losses in loss.py
Add loss computations in trainer.get_representation_loss(...) and trainer.update_c(...)
Add appropriate tests in test_classifier.py (for archi) and test_trainer.py(for losses)