modelscope--ms-swift
wehub-resource-sync
8 行
261 B
Bash
可执行文件
8 行
261 B
Bash
可执行文件
#!/bin/bash
|
|
# Ray-based Megatron GRPO — separate mode
|
|
export CUDA_VISIBLE_DEVICES=${CUDA_VISIBLE_DEVICES:-0,1,2,3,4,5,6,7}
|
|
|
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
|
|
|
megatron rlhf --use_ray true --config "$SCRIPT_DIR/ray_grpo_separate.yaml"
|