pyiqa.utils.logger ================== .. py:module:: pyiqa.utils.logger Module Contents --------------- .. py:data:: initialized_logger .. py:class:: AvgTimer(window=200) .. py:method:: start() .. py:method:: record() .. py:method:: get_current_time() .. py:method:: get_avg_time() .. py:class:: MessageLogger(opt, start_iter=1, tb_logger=None) Message logger for printing. :param opt: Config. It contains the following keys: name (str): Exp name. logger (dict): Contains 'print_freq' (str) for logger interval. train (dict): Contains 'total_iter' (int) for total iters. use_tb_logger (bool): Use tensorboard logger. :type opt: dict :param start_iter: Start iter. Default: 1. :type start_iter: int :param tb_logger (obj: `tb_logger`): Tensorboard logger. Default: None. .. py:method:: reset_start_time() .. py:function:: init_tb_logger(log_dir) .. py:function:: init_wandb_logger(opt) We now only use wandb to sync tensorboard log. .. py:function:: get_root_logger(logger_name='pyiqa', log_level=logging.INFO, log_file=None) Get the root logger. The logger will be initialized if it has not been initialized. By default a StreamHandler will be added. If `log_file` is specified, a FileHandler will also be added. :param logger_name: root logger name. Default: 'basicsr'. :type logger_name: str :param log_file: The log filename. If specified, a FileHandler will be added to the root logger. :type log_file: str | None :param log_level: The root logger level. Note that only the process of rank 0 is affected, while other processes will set the level to "Error" and be silent most of the time. :type log_level: int :returns: The root logger. :rtype: logging.Logger .. py:function:: get_env_info() Get environment information. Currently, only log the software version.