fix

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>

Author: akoumpa

examples/llm_finetune/llama3_2/llama3_2_1b_squad.yaml

@@ -20,7 +20,7 @@
 
 step_scheduler:
   global_batch_size: 64
-  local_batch_size: 1
+  local_batch_size: 8
   ckpt_every_steps: 1000
   val_every_steps: 10  # will run every x number of gradient steps
   num_epochs: 1