Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/ollama.git" did not exist on "7e9405fd0762a8fdbd7a3b3ea7a9268775fd944e"
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
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
d2go/runner/lightning_task.py
View file @ 4731c56a
...@@ -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