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
Add additional dependencies for ModelTrainer #5668
Describe the feature you'd like
When migrating from sagemaker v2 to v3, the Estimator can be replaced by ModelTrainer. However, the tensorboard_output_configanddependencies parameters in Estimator has no equivalent in the ModelTrainer. This functionality would be useful to have in ModelTrainer to maintain feature parity with the v2 Estimator.
How would this feature be used? Please describe.
Allows the import of additional source files and dependencies into the model training container. * Allows debugging visualization using TensorBoard, with customization on the location in Amazon S3 to store the output, as well as the local path in the container. (addressed by ModelTrainer.with_tensorboard_output_config)
changed the title [-]Add TensorBoardOutputConfig for ModelTrainer[/-][+]Add TensorBoardOutputConfig and additional dependencies for ModelTrainer[/+]on Mar 24, 2026
changed the title [-]Add TensorBoardOutputConfig and additional dependencies for ModelTrainer[/-][+]Add ~TensorBoardOutputConfig and~ additional dependencies for ModelTrainer[/+]on Mar 25, 2026
changed the title [-]Add ~TensorBoardOutputConfig and~ additional dependencies for ModelTrainer[/-][+]Add additional dependencies for ModelTrainer[/+]on Mar 25, 2026
Activity
[-]Add TensorBoardOutputConfig for ModelTrainer[/-][+]Add TensorBoardOutputConfig and additional dependencies for ModelTrainer[/+][-]Add TensorBoardOutputConfig and additional dependencies for ModelTrainer[/-][+]Add ~TensorBoardOutputConfig and~ additional dependencies for ModelTrainer[/+][-]Add ~TensorBoardOutputConfig and~ additional dependencies for ModelTrainer[/-][+]Add additional dependencies for ModelTrainer[/+]