mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-09 01:52:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Revert changes to setup_logger. (#468)

Browse Source
This commit is contained in:
Fangjun Kuang 2022-07-08 09:15:37 +08:00 committed by GitHub
parent 8761452a2c
commit e5fdbcd480
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
icefall/utils.py
View File

@ -96,8 +96,6 @@ def str2bool(v):
def setup_logger( def setup_logger(
log_filename: Pathlike, log_filename: Pathlike,
log_level: str = "info", log_level: str = "info",
rank: int = 0,
world_size: int = 1,
use_console: bool = True, use_console: bool = True,
) -> None: ) -> None:
"""Setup log level. """Setup log level.
@ -108,16 +106,14 @@ def setup_logger(
log_level: log_level:
The log level to use, e.g., "debug", "info", "warning", "error", The log level to use, e.g., "debug", "info", "warning", "error",
"critical" "critical"
rank:
Rank of this node in DDP training.
world_size:
Number of nodes in DDP training.
use_console: use_console:
True to also print logs to console. True to also print logs to console.
""" """
now = datetime.now() now = datetime.now()
date_time = now.strftime("%Y-%m-%d-%H-%M-%S") date_time = now.strftime("%Y-%m-%d-%H-%M-%S")
if world_size > 1: if dist.is_available() and dist.is_initialized():
world_size = dist.get_world_size()
rank = dist.get_rank()
formatter = f"%(asctime)s %(levelname)s [%(filename)s:%(lineno)d] ({rank}/{world_size}) %(message)s" # noqa formatter = f"%(asctime)s %(levelname)s [%(filename)s:%(lineno)d] ({rank}/{world_size}) %(message)s" # noqa
log_filename = f"{log_filename}-{date_time}-{rank}" log_filename = f"{log_filename}-{date_time}-{rank}"
else: else: