pull/546/head
lijiaxing 2023-12-20 17:23:45 +08:00
parent 55c7dd513d
commit 2c55bd5768
1 changed files with 1 additions and 1 deletions

View File

@ -38,7 +38,7 @@ ckpt = dict(
# If you want to initialize your model weights from another model, you must set `auto_resume` to False.
# If you want to train from scratch, please set `auto_resume` to False and 'load_ckpt_info' to None.
auto_resume=True,
checkpoint_every=CHECKPOINT_EVERY, # if checkpoint_every is not set, auto checkpoint_every will be executed
checkpoint_every=CHECKPOINT_EVERY,
async_upload=True, # async ckpt upload. (only work for boto3 ckpt)
async_upload_tmp_folder="/dev/shm/internlm_tmp_ckpt/", # path for temporarily files during asynchronous upload.
auto_save_time=1200, # time to control oss_snapshot_freq. If not set, 1200 seconds by default.