Skip to content

Add additional dependencies for ModelTrainer #5668

@wickeat

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

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