mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-06-05 05:32:03 +00:00
3 lines
210 B
Bash
3 lines
210 B
Bash
export NCCL_BLOCKING_WAIT=1
|
|
CUDA_VISIBLE_DEVICES=4,5,6,7 python rl_example.py --dataset /home/share/data/dataset/math_competition_train_short.jsonl --model /home/share/data/model/Qwen2.5-3B -t 1 -i 2 -b vllm
|