From 891b8e7fac993a7fee2dccb5a54ec18f6110d5e0 Mon Sep 17 00:00:00 2001 From: binmakeswell Date: Thu, 6 Apr 2023 18:08:16 +0800 Subject: [PATCH] [chat] fix stage3 PPO sample sh command (#3477) --- applications/Chat/examples/train_prompts.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/applications/Chat/examples/train_prompts.sh b/applications/Chat/examples/train_prompts.sh index db73ac8e8..b750cf358 100755 --- a/applications/Chat/examples/train_prompts.sh +++ b/applications/Chat/examples/train_prompts.sh @@ -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