diff --git a/examples/distributed_training.py b/examples/distributed_training.py index 7030a9f4d8..3b726814f6 100644 --- a/examples/distributed_training.py +++ b/examples/distributed_training.py @@ -92,6 +92,9 @@ def main(): val_dataloader=val_dataloader, val_cfg=dict(), val_evaluator=dict(type=Accuracy), + custom_hooks=[ + dict(type='ProfilerHook', on_trace_ready=dict(type='tb_trace')) + ], launcher=args.launcher, ) runner.train() diff --git a/mmengine/logging/message_hub.py b/mmengine/logging/message_hub.py index c1dda38652..01b283861e 100644 --- a/mmengine/logging/message_hub.py +++ b/mmengine/logging/message_hub.py @@ -173,7 +173,6 @@ def update_scalars(self, log_dict: dict, resumed: bool = True) -> None: assert isinstance(log_dict, dict), ('`log_dict` must be a dict!, ' f'but got {type(log_dict)}') for log_name, log_val in log_dict.items(): - self._set_resumed_keys(log_name, resumed) if isinstance(log_val, dict): assert 'value' in log_val, \ f'value must be defined in {log_val}' @@ -232,7 +231,6 @@ def update_info_dict(self, info_dict: dict, resumed: bool = True) -> None: assert isinstance(info_dict, dict), ('`log_dict` must be a dict!, ' f'but got {type(info_dict)}') for key, value in info_dict.items(): - self._set_resumed_keys(key, resumed) self.update_info(key, value, resumed=resumed) def _set_resumed_keys(self, key: str, resumed: bool) -> None: