From 29b4c95388812f445d1e9312f9a34af50f59b331 Mon Sep 17 00:00:00 2001 From: Terry Kong Date: Wed, 16 Apr 2025 23:37:05 -0700 Subject: [PATCH] fix: grpo func test 10 step -> 3 step to speed up CI Signed-off-by: Terry Kong --- tests/functional/grpo.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/functional/grpo.sh b/tests/functional/grpo.sh index 497a240d84..b61442227b 100755 --- a/tests/functional/grpo.sh +++ b/tests/functional/grpo.sh @@ -19,7 +19,7 @@ mkdir -p $LOG_DIR cd $PROJECT_ROOT python -u $PROJECT_ROOT/examples/run_grpo_math.py \ cluster.gpus_per_node=2 \ - grpo.max_num_steps=10 \ + grpo.max_num_steps=3 \ logger.tensorboard_enabled=true \ logger.log_dir=$LOG_DIR \ logger.wandb_enabled=false \