Skip to content

Commit 7ae43c8

Browse files
Aditi2424adishaaSharma
authored
Fix HyperparameterTuner to include ModelTrainer internal channels (#5516)
* Fix HyperparameterTuner to include ModelTrainer internal channels HyperparameterTuner._build_training_job_definition() was only copying user-provided input channels, missing ModelTrainer's internal channels (code, sm_drivers). This caused tuning jobs to fail when using custom training scripts. - Add logic to include ModelTrainer's input_data_config channels - Refactor test_tuner.py with factory functions to reduce duplication - Add test for internal channels inclusion Fixes #5508 * fix formatting * Fix static_hyperparameters initialization and reference --------- Co-authored-by: adishaa <adishaa@amazon.com> Co-authored-by: Sharma <adishaa@14147d9496bd.ant.amazon.com>
1 parent ea3ea86 commit 7ae43c8

File tree

2 files changed

+288
-174
lines changed

2 files changed

+288
-174
lines changed

0 commit comments

Comments
 (0)