colossalai.logging.logging

class colossalai.logging.logging.DistributedLogger(name)

This is a distributed event logger class essentially based on logging.

Parameters

name (str) – The name of the logger

static get_instance(name)

Get the unique single logger instance based on name.

Parameters

name (str) – The name of the logger

Returns

A DistributedLogger object

Return type

DistributedLogger

set_level(level)

Set the logging level

Parameters

level (str) – Can only be INFO, DEBUG, WARNING and ERROR

log_to_file(path, mode='a', level='INFO', suffix=None)

Save the logs to file

Parameters

  • path (A string or pathlib.Path object) – The file to save the log

  • mode (str) – The mode to write log into the file

  • level (str) – Can only be INFO, DEBUG, WARNING and ERROR

  • suffix (str) – The suffix string of log’s name

info(message, parallel_mode=ParallelMode.GLOBAL, ranks=None)

Log an info message.

Parameters
warning(message, parallel_mode=ParallelMode.GLOBAL, ranks=None)

Log a warning message.

Parameters
debug(message, parallel_mode=ParallelMode.GLOBAL, ranks=None)

Log a debug message.

Parameters
error(message, parallel_mode=ParallelMode.GLOBAL, ranks=None)

Log an error message.

Parameters