github
/
InternLM
mirror of https://github.com/InternLM/InternLM
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

feat(utils/writer.py): add class comment

pull/63/head
黄婷 2023-07-12 19:48:04 +08:00
parent 05bfe7b292
commit 6dd6d07813
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
internlm/utils/writer.py
View File

@ -95,13 +95,19 @@ class Writer:
Customed writer based on tensorboard for recording training metrics. Customed writer based on tensorboard for recording training metrics.
Args: Args:
launch_time (str): A string representing the launch time of the training.
tensorboard_folder (str): A string representing the folder for saving tensorboard logs.
resume_tb_folder (str): A string representing the folder for resuming tensorboard logs.
step_count (int): An integer representing the step count of the training.
config (str): A string representing the configuration of the training.
logger (logging.Logger): A logging.Logger object for logging information during training.
enable_tb (bool): A boolean indicating whether to enable the tensorboard writer.
Return:
""" """
def __init__( def __init__(
self, self,
launch_time, launch_time: str,
tensorboard_folder: str = None, tensorboard_folder: str = None,
resume_tb_folder: str = None, resume_tb_folder: str = None,
step_count: int = 0, step_count: int = 0,