Commit 4731c56a authored by Kai Zhang's avatar Kai Zhang Committed by Facebook GitHub Bot
Browse files

StyleGAN2

Summary: A Lightning task for training StyleGAN2.

Reviewed By: tax313

Differential Revision: D28922408

fbshipit-source-id: bdc9e7370de1b7b7ca9086bc6c0acbe66810d5f8
parent 3805abbe
...@@ -200,7 +200,7 @@ class DefaultTask(pl.LightningModule): ...@@ -200,7 +200,7 @@ class DefaultTask(pl.LightningModule):
for idx, evaluator in enumerate(evaluators): for idx, evaluator in enumerate(evaluators):
dataset_name = self.cfg.DATASETS.TEST[idx] dataset_name = self.cfg.DATASETS.TEST[idx]
res[dataset_name] = evaluator.evaluate() res[dataset_name] = evaluator.evaluate()
nested_res[tag.value] = res nested_res[tag.lower()] = res
self.eval_res = nested_res self.eval_res = nested_res
flattened = pl.loggers.LightningLoggerBase._flatten_dict(nested_res) flattened = pl.loggers.LightningLoggerBase._flatten_dict(nested_res)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment