Skip to content

Add additional dependencies for ModelTrainer #5668

Open

Description

@wickeat

Describe the feature you'd like
When migrating from sagemaker v2 to v3, the Estimator can be replaced by ModelTrainer. However, the tensorboard_output_configand dependencies 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)

Describe alternatives you've considered
None

Additional context
None

Activity

  1. changed the title [-]Add TensorBoardOutputConfig for ModelTrainer[/-] [+]Add TensorBoardOutputConfig and additional dependencies for ModelTrainer[/+] on Mar 24, 2026
  2. changed the title [-]Add TensorBoardOutputConfig and additional dependencies for ModelTrainer[/-] [+]Add ~TensorBoardOutputConfig and~ additional dependencies for ModelTrainer[/+] on Mar 25, 2026
  3. changed the title [-]Add ~TensorBoardOutputConfig and~ additional dependencies for ModelTrainer[/-] [+]Add additional dependencies for ModelTrainer[/+] on Mar 25, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions