From 11770a68cfefedeabbb21f8a3049bf960547b3f3 Mon Sep 17 00:00:00 2001 From: binmakeswell Date: Thu, 6 Apr 2023 18:05:41 +0800 Subject: [PATCH] [coati] fix stage3 PPO sample sh command --- 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 db73ac8e8e85..b750cf3581a6 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