From 37e0c86e5ac10de0ff88df53606455814c76976a Mon Sep 17 00:00:00 2001 From: Guoteng <32697156+SolenoidWGT@users.noreply.github.com> Date: Fri, 13 Oct 2023 16:46:14 +0800 Subject: [PATCH] fix(init): allow resume_tb_folder is an empty string (#391) --- internlm/utils/writer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/internlm/utils/writer.py b/internlm/utils/writer.py index b519b95..fb41fe5 100644 --- a/internlm/utils/writer.py +++ b/internlm/utils/writer.py @@ -42,8 +42,8 @@ def init_tb_writer( # dir of the last task by 'make_launch_script.sh'. # If we load ckpt, 'resume_tb_folder' will be overwritten as the # reloaded 'train_state.resume_tb_folder'.s - if resume_tb_folder is not None: - assert len(resume_tb_folder) > 0 and resume_tb_folder != "/" + if resume_tb_folder is not None and len(resume_tb_folder) > 0: + assert resume_tb_folder != "/" if not os.path.exists(resume_tb_folder): logger.error( f"Can't found resume_tb_folder{resume_tb_folder}, \