Method __init__

utils/logger.py:24–63

tensorboard logger Args: place (str, optional): place to save tb logging. Defaults to './tb_runs/'. file_logger_name (str, optional): name of file logger. Defaults to None. random_id: (bool, optional): append id after tensorboard dir. Defaults to True

(self, place="./runs/", file_dir='./logs/', file_logger_name=None, random_id=True, tb_comment=None,
                 **tb_kwargs)

class TensorboardLogger:
    def __init__(self, place="./runs/", file_dir='./logs/', file_logger_name=None, random_id=True, tb_comment=None,
                 **tb_kwargs) -> None:
        """tensorboard logger

        Args:
            place (str, optional): place to save tb logging. Defaults to './tb_runs/'.
            file_logger_name (str, optional): name of file logger. Defaults to None.
            random_id: (bool, optional): append id after tensorboard dir. Defaults to True
            tb_kwargs: kwargs for tensorboardX.writer.SummaryWriter
                comment: Optional[str] = "",
                purge_step: Optional[int] = None,
                max_queue: Optional[int] = 10,
                flush_secs: Optional[int] = 120,
                filename_suffix: Optional[str] = '',
                write_to_disk: Optional[bool] = True,
                log_dir: Optional[str] = None,
                comet_config: Optional[dict] = {"disabled": True}, 
        """
        if not place_exists(place):
            os.mkdir(place)
            print("Created directory: ", place)

        if random_id:
            # use random id as dir name
            id = generate_id(8) + '' if tb_comment is None else '_{}'.format(tb_comment)
            place = os.path.join(place, id)
            os.mkdir(place)
        else:
            # use time as dir name
            stf_time = time.strftime("%m-%d %H:%M", time.localtime())
            name = stf_time + '' if tb_comment is None else '_{}'.format(tb_comment)
            place = os.path.join(place, name)
            os.mkdir(place)

        self.writer = writer.SummaryWriter(place, **tb_kwargs)

        assert file_logger_name is not None, '@file_logger_name should be set'
        self.file_writer = PrintLogger(os.path.join(file_dir, file_logger_name + '.log'))

        self._tb_print = partial(self.file_writer.log, level=logging.DEBUG)

    def print(self, msg: str, level=logging.INFO):
        self.file_writer.log(msg, level)