aws / aws/sagemaker-python-sdk
Align Remote function classes to ModelTrainer: Include training_plan_arn to remote function classes @remote decorator and RemoteExecutor
- 主要语言
- Python
- 星标
- 2.3k
- 派生
- 1.3k
- 平均合并
- 1 天 22 小时
- 30 天内合并 PR
- 35
描述
**Describe the feature you'd like**
A clear and concise description of the functionality you want.
ModelTrainer and Estimator classes currently support training_plan_arn, in order to run training workload by leveraging the instances included in the plan. This functionality is not currently available with remote function classes, which is adopted by customer to start prototyping workloads on Amazon SageMaker.
**How would this feature be used? Please describe.**
A clear and concise description of the use case for this feature. Please provide an example, if possible.
```
@remote(
...
training_plan_arn="arn.1234....."
)
def train(args):
pass
```
**Describe alternatives you've considered**
A clear and concise description of any alternative solutions or features you've considered.
There are no alternative solutions
**Additional context**
Add any other context or screenshots about the feature request here.
贡献指南
调研方向
首先比较 ModelTrainer 和 Estimator 接受并使用 training_plan_arn 的方式,然后跟踪 @remote decorator 和 RemoteExecutor 在等效配置下的路径。定义 ARN 从 decorator 参数进入远程执行的预期流程,并添加覆盖,展示 remote function 可以使用指定的 training plan 启动。
由索引模型根据 Issue 内容生成。
评估
- 技术栈
- aws, python
- 领域
- cloud, machine-learning
- Issue 类型
- 功能
- 难度
- 4/5
- 预计耗时
- 3-5 天
- 活跃度
- 停滞
- 描述清晰度
- 基本清楚
- 新手友好度
- 45/100