aws / aws/sagemaker-python-sdk
Align Remote function classes to ModelTrainer: Include training_plan_arn to remote function classes @remote decorator and RemoteExecutor
- Dominant language
- Python
- Stars
- 2.3k
- Forks
- 1.3k
- Avg merge
- 2d 2h
- Merged PRs (30d)
- 32
Description
**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.
Contributor guide
Assessment
This issue has not been assessed yet.