[recipe] fix: Correct Ministral 3 SFT learning rates (#5092)

Signed-off-by: Yu Yao <yaoyu.094@gmail.com>

Author: yaoyu-33

src/megatron/bridge/recipes/ministral3/h100/ministral3.py

@@ -94,8 +94,8 @@ def ministral3_3b_sft_1gpu_h100_bf16_config() -> ConfigContainer:
     opt_cfg, scheduler_cfg = distributed_fused_adam_with_cosine_annealing(
         lr_warmup_iters=10,
         lr_decay_iters=50,
-        max_lr=0.00005,
-        min_lr=0.000005,
+        max_lr=0.000005,
+        min_lr=0.0000005,
     )
     cfg.optimizer = opt_cfg
     cfg.scheduler = scheduler_cfg
@@ -209,8 +209,8 @@ def ministral3_8b_sft_2gpu_h100_bf16_config() -> ConfigContainer:
     opt_cfg, scheduler_cfg = distributed_fused_adam_with_cosine_annealing(
         lr_warmup_iters=10,
         lr_decay_iters=50,
-        max_lr=0.00005,
-        min_lr=0.000005,
+        max_lr=0.000005,
+        min_lr=0.0000005,
     )
     cfg.optimizer = opt_cfg
     cfg.scheduler = scheduler_cfg
@@ -324,8 +324,8 @@ def ministral3_14b_sft_4gpu_h100_bf16_config() -> ConfigContainer:
     opt_cfg, scheduler_cfg = distributed_fused_adam_with_cosine_annealing(
         lr_warmup_iters=10,
         lr_decay_iters=50,
-        max_lr=0.00005,
-        min_lr=0.000005,
+        max_lr=0.000005,
+        min_lr=0.0000005,
     )
     cfg.optimizer = opt_cfg
     cfg.scheduler = scheduler_cfg

tests/unit_tests/recipes/test_ministral3_recipes.py

@@ -125,6 +125,19 @@ def test_each_ministral3_sft_recipe_builds_config(recipe_func: Callable, monkeyp
     assert cfg.peft is None
 
 
+@pytest.mark.parametrize("recipe_func", _MINISTRAL3_SFT_FUNCS)
+def test_each_ministral3_sft_recipe_uses_recommended_learning_rate(
+    recipe_func: Callable, monkeypatch: pytest.MonkeyPatch
+):
+    """Test that each Ministral3 full-SFT recipe uses its recommended learning-rate range."""
+    patch_recipe_module_global(monkeypatch, _ministral3_module, "AutoBridge", _FakeAutoBridge)
+
+    cfg = recipe_func()
+
+    assert cfg.optimizer.lr == pytest.approx(5e-6)
+    assert cfg.optimizer.min_lr == pytest.approx(5e-7)
+
+
 @pytest.mark.parametrize("recipe_func", _MINISTRAL3_PEFT_FUNCS)
 def test_each_ministral3_peft_recipe_builds_config(recipe_func: Callable, monkeypatch: pytest.MonkeyPatch):
     """Test that each Ministral3 PEFT recipe function builds a valid configuration."""