From 89388810def6df1b484c820cf8af87b132a973f5 Mon Sep 17 00:00:00 2001 From: kailash Date: Tue, 21 Jul 2026 06:11:38 +0000 Subject: [PATCH] Update batch size for mbridge qwen3-32B on MI300X --- .../configs/MI300X/qwen3_32b_lora_posttrain.yaml | 2 +- .../megatron_bridge/configs/MI300X/qwen3_32b_sft_posttrain.yaml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/megatron_bridge/configs/MI300X/qwen3_32b_lora_posttrain.yaml b/examples/megatron_bridge/configs/MI300X/qwen3_32b_lora_posttrain.yaml index 330c9e382..b1493908c 100644 --- a/examples/megatron_bridge/configs/MI300X/qwen3_32b_lora_posttrain.yaml +++ b/examples/megatron_bridge/configs/MI300X/qwen3_32b_lora_posttrain.yaml @@ -31,7 +31,7 @@ modules: # Training configuration train_iters: 200 global_batch_size: 32 - micro_batch_size: 2 + micro_batch_size: 1 seq_length: 8192 eval_interval: 30 save_interval: 50 diff --git a/examples/megatron_bridge/configs/MI300X/qwen3_32b_sft_posttrain.yaml b/examples/megatron_bridge/configs/MI300X/qwen3_32b_sft_posttrain.yaml index 75c721ed2..f05718454 100644 --- a/examples/megatron_bridge/configs/MI300X/qwen3_32b_sft_posttrain.yaml +++ b/examples/megatron_bridge/configs/MI300X/qwen3_32b_sft_posttrain.yaml @@ -31,7 +31,7 @@ modules: # Training configuration train_iters: 200 global_batch_size: 8 - micro_batch_size: 2 + micro_batch_size: 1 seq_length: 8192 eval_interval: 30 save_interval: 50