Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
SeaOfOcean committed Aug 28, 2024
1 parent 53e98d5 commit ab86b67
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/zh/tutorial/tutorial_llama2.md
Original file line number Diff line number Diff line change
Expand Up @@ -70,7 +70,7 @@ export model_size=llama2-7B
LOAD_PATH=$MEGATRON_LLAMA2_CKPT_PATH \
TOKENIZER_MODEL=$LLAMA2_TOKENIZER_MODEL \
DATASET_PATH=$DATASET_ROOT/sft/ \
bash scripts/llama2_sft.sh
bash scripts/train_sft_llama.sh
```

训练 log 和训练完成的模型默认会存放在`${CHATLEARN}/output/sft`中,可以通过 CHECKPOINT_PATH 来指定模型保存路径,具体的定义详见`${CHATLEARN}/examples/megatron/scripts/train_sft_llama.sh`脚本。
Expand Down

0 comments on commit ab86b67

Please sign in to comment.