Skip to content

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

@brunopistone

Description

@brunopistone

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.

Metadata

Metadata

Assignees

No one assigned

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions