diff --git a/mmengine/config/config.py b/mmengine/config/config.py index 8d3dc6d40f243827bfca2402ad0d54b1c5d978d6..e29260a33b6c22be77455f30412f558fba8c8da6 100644 --- a/mmengine/config/config.py +++ b/mmengine/config/config.py @@ -18,6 +18,7 @@ from addict import Dict from yapf.yapflib.yapf_api import FormatCode from mmengine.fileio import dump, load +from mmengine.logging import print_log from mmengine.utils import (check_file_exist, get_installed_path, import_modules_from_strings, is_installed) from .utils import (RemoveAssignFromAST, _get_external_cfg_base_path, @@ -86,7 +87,9 @@ def add_args(parser: ArgumentParser, parser.add_argument( '--' + prefix + k, type=type(next(iter(v))), nargs='+') else: - print(f'cannot parse key {prefix + k} of type {type(v)}') + print_log( + f'cannot parse key {prefix + k} of type {type(v)}', + logger='current') return parser diff --git a/mmengine/optim/scheduler/param_scheduler.py b/mmengine/optim/scheduler/param_scheduler.py index df2e25da12009cee2bbc036ca6fef33214a90381..7b4ca4ffaff765c63bfb00d335a39e7d9c06fa48 100644 --- a/mmengine/optim/scheduler/param_scheduler.py +++ b/mmengine/optim/scheduler/param_scheduler.py @@ -13,6 +13,7 @@ from typing import Callable, List, Optional, Sequence, Union from torch.optim import Optimizer +from mmengine.logging import print_log from mmengine.optim import OptimWrapper from mmengine.registry import PARAM_SCHEDULERS @@ -172,8 +173,9 @@ class _ParamScheduler: value (float): The parameter value. """ if is_verbose: - print('Adjusting parameter value' - ' of group {} to {:.4e}.'.format(group, value)) + print_log( + f'Adjusting parameter value of group {group} to {value:.4e}.', + logger='current') def step(self): """Adjusts the parameter value of each parameter group based on the diff --git a/mmengine/registry/utils.py b/mmengine/registry/utils.py index 5184e2ebedbba5dd15060da61cffea09709e52b7..7ce11da1557e9984c9c9aca4510156a6f2ec0835 100644 --- a/mmengine/registry/utils.py +++ b/mmengine/registry/utils.py @@ -4,6 +4,7 @@ import os.path as osp from typing import Optional from mmengine.fileio import dump +from mmengine.logging import print_log from . import root from .registry import Registry @@ -35,8 +36,10 @@ def traverse_registry_tree(registry: Registry, verbose: bool = True) -> list: else: registry_info[folder] = [name] if verbose: - print(f"Find {num_modules} modules in {scope}'s " - f"'{_registry.name}' registry ") + print_log( + f"Find {num_modules} modules in {scope}'s " + f"'{_registry.name}' registry ", + logger='current') modules_info.append(registry_info) else: return @@ -80,9 +83,10 @@ def count_registered_modules(save_path: Optional[str] = None, scan_date=datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S'), registries=registries_info) if verbose: - print('Finish registry analysis, got: ', scan_data) + print_log( + f'Finish registry analysis, got: {scan_data}', logger='current') if save_path is not None: json_path = osp.join(save_path, 'modules_statistic_results.json') dump(scan_data, json_path, indent=2) - print(f'Result has been saved to {json_path}') + print_log(f'Result has been saved to {json_path}', logger='current') return scan_data