Update 7B_sft.py

pull/380/head
Guoteng 2023-09-27 21:38:45 +08:00 committed by GitHub
parent 3d8690bf8b
commit 5757a88ca4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

@ -36,7 +36,7 @@ ckpt = dict(
# Please be aware that if `auto_resume` is not set (its default value is True), it will not load the checkpoint # Please be aware that if `auto_resume` is not set (its default value is True), it will not load the checkpoint
# path specified in `load_ckpt_info` by default. # path specified in `load_ckpt_info` by default.
# If you want to initialize your model weights from another model, you must set `auto_resume` to False. # 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 load_ckpt_info=None. # If you want to train from scratch, please set `auto_resume` to False and 'load_ckpt_info' to None.
auto_resume=False, auto_resume=False,
checkpoint_every=CHECKPOINT_EVERY, checkpoint_every=CHECKPOINT_EVERY,
async_upload=True, # async ckpt upload. (only work for boto3 ckpt) async_upload=True, # async ckpt upload. (only work for boto3 ckpt)