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

fix(ci): fix train error (#228) 

Co-authored-by: huangting4201 <huangting3@sensetime.com>
pull/230/head
huangting4201 2023-08-24 17:11:32 +08:00 committed by GitHub
parent 7c820cfa40
commit 04c02a61b2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
ci_scripts/train/ci_7B_sft.py
View File

@ -15,6 +15,7 @@ VOCAB_SIZE = 103168
SAVE_CKPT_FOLDER = "local:llm_ckpts" SAVE_CKPT_FOLDER = "local:llm_ckpts"
# LOAD_CKPT_FOLDER = "local:llm_ckpts/49" # LOAD_CKPT_FOLDER = "local:llm_ckpts/49"
ckpt = dict( ckpt = dict(
enable_save_ckpt=True,
# Path to save training ckpt. # Path to save training ckpt.
save_ckpt_folder=SAVE_CKPT_FOLDER, save_ckpt_folder=SAVE_CKPT_FOLDER,
# Path to continue training ckpt (load model weights and scheduler/context states). # Path to continue training ckpt (load model weights and scheduler/context states).