-
Notifications
You must be signed in to change notification settings - Fork 1.2k
Open
Description
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)
Describe alternatives you've considered
None
Additional context
None
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels