File size: 805 Bytes
e791b16 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 | #!/usr/bin/env bash
# Launch rollout workers using Ray
# Example usage:
# scripts/launch_rollout.sh <num_workers> <model_path>
NUM_WORKERS=$1
MODEL_PATH=$2
if [ -z "$NUM_WORKERS" ] || [ -z "$MODEL_PATH" ]; then
echo "Usage: $0 <num_workers> <model_path>"
exit 1
fi
# Start Ray head (if not already running)
if ! ray status > /dev/null 2>&1; then
ray start --head
fi
# Launch rollout actors
for i in $(seq 1 $NUM_WORKERS); do
ray exec "import src.rollout.async_engine as ae; from src.buffer.replay_buffer import BoundedReplayBuffer; buffer = BoundedReplayBuffer(); engine = ae.AsyncEngine(engine=None, buffer=buffer);" --address=auto &
# The above is a placeholder; replace with actual Ray actor creation script.
done
echo "Launched $NUM_WORKERS rollout workers for model at $MODEL_PATH"
|