diff --git a/examples/llm_finetune/nemotron/nemotron_super_v3_hellaswag.yaml b/examples/llm_finetune/nemotron/nemotron_super_v3_hellaswag.yaml index 3c03bd6351..36ec56283f 100644 --- a/examples/llm_finetune/nemotron/nemotron_super_v3_hellaswag.yaml +++ b/examples/llm_finetune/nemotron/nemotron_super_v3_hellaswag.yaml @@ -111,5 +111,6 @@ ci: tokenizer_name: nvidia/NVIDIA-Nemotron-3-Super-120B-A12B-BF16 hf_device_map_auto: true no_check_resume: true + step_scheduler.global_batch_size: 64 dataset.num_samples_limit: 500 validation_dataset.num_samples_limit: 500