Skip to content

Commit 604f403

Browse files
authored
fix: Speed up DPO functional test (#241)
Signed-off-by: ashors1 <ashors@nvidia.com>
1 parent 2b39fb0 commit 604f403

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

tests/functional/dpo.sh

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,8 @@ python -u $PROJECT_ROOT/examples/run_dpo.py \
2222
cluster.gpus_per_node=2 \
2323
dpo.max_num_steps=3 \
2424
dpo.val_batches=1 \
25+
dpo.val_global_batch_size=8 \
26+
policy.train_global_batch_size=8 \
2527
logger.tensorboard_enabled=true \
2628
logger.log_dir=$LOG_DIR \
2729
logger.wandb_enabled=false \

0 commit comments

Comments
 (0)