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

fix(init): allow resume_tb_folder is an empty string

pull/391/head
877825076@qq.com 2023-10-07 13:28:07 +08:00
parent 4f9e8cd70d
commit 1d7d774ecf
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
internlm/utils/writer.py
View File

@ -42,8 +42,8 @@ def init_tb_writer(
# dir of the last task by 'make_launch_script.sh'. # dir of the last task by 'make_launch_script.sh'.
# If we load ckpt, 'resume_tb_folder' will be overwritten as the # If we load ckpt, 'resume_tb_folder' will be overwritten as the
# reloaded 'train_state.resume_tb_folder'.s # reloaded 'train_state.resume_tb_folder'.s
if resume_tb_folder is not None: if resume_tb_folder is not None and len(resume_tb_folder) > 0:
assert len(resume_tb_folder) > 0 and resume_tb_folder != "/" assert resume_tb_folder != "/"
if not os.path.exists(resume_tb_folder): if not os.path.exists(resume_tb_folder):
logger.error( logger.error(
f"Can't found resume_tb_folder{resume_tb_folder}, \ f"Can't found resume_tb_folder{resume_tb_folder}, \