aws / aws/sagemaker-python-sdk

Align Remote function classes to ModelTrainer: Include training_plan_arn to remote function classes @remote decorator and RemoteExecutor

未关闭
#5,157 0 条评论 0 个 reaction 已指派 0 人 在 GitHub 查看
remote-function type: feature request
主要语言
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

把新 issue 发到你的邮箱

精选适合新手参与的 GitHub issue 摘要。