diff --git a/configs/7B_sft.py b/configs/7B_sft.py index 7e81f42..4358ae8 100644 --- a/configs/7B_sft.py +++ b/configs/7B_sft.py @@ -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.