diff --git a/src/sagemaker/hyperpod/cli/recipe_param_order.py b/src/sagemaker/hyperpod/cli/recipe_param_order.py index 0eec314a..aed682ed 100644 --- a/src/sagemaker/hyperpod/cli/recipe_param_order.py +++ b/src/sagemaker/hyperpod/cli/recipe_param_order.py @@ -49,7 +49,7 @@ ("max_steps", "Core Hyperparameters"), # Advanced Hyperparameters (includes technique-specific params) - ("lr_warmup_ratio", "Advanced Hyperparameters"), + ("lr_warmup_steps_ratio", "Advanced Hyperparameters"), ("max_context_length", "Advanced Hyperparameters"), ("max_prompt_length", "Advanced Hyperparameters"), ("max_length", "Advanced Hyperparameters"), diff --git a/test/integration_tests/init/test_recipe_job_creation.py b/test/integration_tests/init/test_recipe_job_creation.py index 5e6743be..a0f89b27 100644 --- a/test/integration_tests/init/test_recipe_job_creation.py +++ b/test/integration_tests/init/test_recipe_job_creation.py @@ -97,7 +97,7 @@ def test_configure_recipe_job(runner, job_name, test_directory): "--data-path", "/data/recipes-data/sft/zc_train_256.jsonl", "--global-batch-size", "8", "--learning-rate", "0.0001", - "--lr-warmup-ratio", "0.1", + "--lr-warmup-steps-ratio", "0.1", "--max-epochs", "20", "--output-path", "/data/output/qwen3-sft", "--results-directory", "/data/results/qwen3-sft",