mirror of https://github.com/InternLM/InternLM
Update 7B_sft.py
parent
3d8690bf8b
commit
5757a88ca4
|
@ -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)
|
||||||
|
|
Loading…
Reference in New Issue