Skip to content

GitLab

Unverified Commit 891b8e7f authored by binmakeswell's avatar binmakeswell Committed by GitHub
Browse files

[chat] fix stage3 PPO sample sh command (#3477)

parent c701b77b
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
applications/Chat/examples/train_prompts.sh
View file @ 891b8e7f
......@@ -15,4 +15,4 @@ set_n_least_used_CUDA_VISIBLE_DEVICES() {
set_n_least_used_CUDA_VISIBLE_DEVICES 2
torchrun --standalone --nproc_per_node=2 train_prompts.py prompts.csv --strategy colossalai_zero2
torchrun --standalone --nproc_per_node=2 train_prompts.py --prompt_path /path/to/data.json --strategy colossalai_zero2
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment