You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
{{ message }}
Repository navigation
Commit 8a33376
Browse filesBrowse the repository at this point in the historyBrowse files
feat(train): add list_hyperparameters() for pre-trainer HP discovery
Add a public utility function that returns available hyperparameters
for a model/technique/training_type combination without requiring a
fully constructed trainer object.
This enables tools and scripts to discover valid hyperparameter names,
defaults, and ranges before setting up training infrastructure (model
package groups, datasets, roles, etc.).
Motivation: COE 398545 identified that hardcoded HP names in downstream
consumers break when recipe templates rename parameters. Dynamic
discovery at code-generation time prevents this class of failure.
Usage:
from sagemaker.train import list_hyperparameters
hp = list_hyperparameters('model-name', 'SFT', 'LORA')
hp.get_info() # display all params
hp.get_info('learning_rate') # display one param
0 commit comments