"vscode:/vscode.git/clone" did not exist on "3b5b1c56983004ca1ee4190d0eb65f98b0101d39"
Commit 6791682f authored by Ning Li (Seattle)'s avatar Ning Li (Seattle) Committed by Facebook GitHub Bot
Browse files

Decouple utilities from `LightningLoggerBase` (#11484)

Summary:
### New commit log messages
- [115a5d08e Decouple utilities from `LightningLoggerBase` (#11484)](https://github.com/PyTorchLightning/pytorch-lightning/pull/11484)

Reviewed By: tangbinh, wat3rBro

Differential Revision: D33960185

fbshipit-source-id: 6be72ad49f8433be6f238b36aa82d3f1b655e6f0
parent 2f0e1c92
......@@ -40,6 +40,7 @@ from detectron2.solver import (
build_optimizer as d2_build_optimizer,
)
from pytorch_lightning.utilities import rank_zero_only, rank_zero_info
from pytorch_lightning.utilities.logger import _flatten_dict
_STATE_DICT_KEY = "state_dict"
_OLD_STATE_DICT_KEY = "model"
......@@ -222,7 +223,7 @@ class DefaultTask(pl.LightningModule):
nested_res[tag.lower()] = res
self.eval_res = nested_res
flattened = pl.loggers.LightningLoggerBase._flatten_dict(nested_res)
flattened = _flatten_dict(nested_res)
if self.trainer.global_rank:
assert (
......
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