From bd96e708ebcc293140549be7753f0ad48b698ca3 Mon Sep 17 00:00:00 2001 From: Ajay Balasa Date: Fri, 22 May 2026 09:46:43 -0700 Subject: [PATCH] Update training script paths in BERT and T5 YAML configurations to reflect new directory structure --- tests/test_utils/recipes/h100/bert.yaml | 2 +- tests/test_utils/recipes/h100/t5.yaml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_utils/recipes/h100/bert.yaml b/tests/test_utils/recipes/h100/bert.yaml index 89499f93c5e..a27f79ee9d9 100644 --- a/tests/test_utils/recipes/h100/bert.yaml +++ b/tests/test_utils/recipes/h100/bert.yaml @@ -45,7 +45,7 @@ spec: "TENSORBOARD_PATH={assets_dir}/tensorboard" "CHECKPOINT_SAVE_PATH={artifacts_dir}/checkpoints" "CHECKPOINT_LOAD_PATH=/mnt/artifacts" - "TRAINING_SCRIPT_PATH=pretrain_bert.py" + "TRAINING_SCRIPT_PATH=examples/bert/pretrain_bert.py" "TRAINING_PARAMS_PATH=./tests/functional_tests/test_cases/{model}/{test_case}/model_config.yaml" "GOLDEN_VALUES_PATH=./tests/functional_tests/test_cases/{model}/{test_case}/golden_values_{environment}_{platforms}.json" "N_REPEAT={n_repeat}" diff --git a/tests/test_utils/recipes/h100/t5.yaml b/tests/test_utils/recipes/h100/t5.yaml index 1761cd3f1e6..619028ef9f4 100644 --- a/tests/test_utils/recipes/h100/t5.yaml +++ b/tests/test_utils/recipes/h100/t5.yaml @@ -45,7 +45,7 @@ spec: "TENSORBOARD_PATH={assets_dir}/tensorboard" "CHECKPOINT_SAVE_PATH={artifacts_dir}/checkpoints" "CHECKPOINT_LOAD_PATH=/mnt/artifacts" - "TRAINING_SCRIPT_PATH=pretrain_t5.py" + "TRAINING_SCRIPT_PATH=examples/t5/pretrain_t5.py" "TRAINING_PARAMS_PATH=./tests/functional_tests/test_cases/{model}/{test_case}/model_config.yaml" "GOLDEN_VALUES_PATH=./tests/functional_tests/test_cases/{model}/{test_case}/golden_values_{environment}_{platforms}.json" "N_REPEAT={n_repeat}"