StyleGAN2

Summary: A Lightning task for training StyleGAN2.

Reviewed By: tax313

Differential Revision: D28922408

fbshipit-source-id: bdc9e7370de1b7b7ca9086bc6c0acbe66810d5f8

d2go/runner/lightning_task.py

@@ -200,7 +200,7 @@ class DefaultTask(pl.LightningModule):
             for idx, evaluator in enumerate(evaluators):
                 dataset_name = self.cfg.DATASETS.TEST[idx]
                 res[dataset_name] = evaluator.evaluate()
-            nested_res[tag.value] = res
+            nested_res[tag.lower()] = res
 
         self.eval_res = nested_res
         flattened = pl.loggers.LightningLoggerBase._flatten_dict(nested_res)