aws / aws/sagemaker-python-sdk
Add additional dependencies for ModelTrainer
- Ngôn ngữ chính
- Python
- Star
- 2.3k
- Fork
- 1.3k
- Merge trung bình
- 1 ngày 22 giờ
- Pull request đã merge (30 ngày)
- 35
Mô tả
**Describe the feature you'd like**
When migrating from sagemaker v2 to v3, the [Estimator](https://sagemaker.readthedocs.io/en/v2.256.0/api/training/estimators.html?highlight=tensorboardoutputconfig#sagemaker.estimator.Estimator) can be replaced by ModelTrainer. However, the ~~`tensorboard_output_config`and~~ `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](https://sagemaker.readthedocs.io/en/v2.253.1/api/training/model_trainer.html#sagemaker.modules.train.model_trainer.ModelTrainer.with_tensorboard_output_config))
**Describe alternatives you've considered**
None
**Additional context**
None
Hướng dẫn đóng góp
Hướng nghiên cứu
Bắt đầu từ entry point của ModelTrainer và so sánh cách xử lý của nó với tham số `dependencies` của v2 Estimator. Theo dõi cách các tệp mã nguồn bổ sung và các dependency đi vào training container, sau đó xác minh rằng ModelTrainer hỗ trợ các import được yêu cầu và giữ nguyên hành vi training đã được ghi trong tài liệu.
Do mô hình lập chỉ mục viết ra từ nội dung của issue.
Đánh giá
- Công nghệ
- aws, python
- Lĩnh vực
- machine-learning
- Loại issue
- Tính năng
- Độ khó
- 3/5
- Thời gian dự kiến
- 1-2 ngày
- Mức độ hoạt động
- Ít trao đổi
- Độ rõ ràng
- Khá rõ ràng
- Mức phù hợp với người mới
- 55/100