Feature Request: Cannot pass output path and ECR Image ARN from Execution Input

[mochacafe]

trafficstars

While creating the Sagemaker training step I have fix the output path location and ECR Image for all the executions. Is there a way we can pass the model output path from the execution input?

[shunjd]

Hi @mochacafe, could you provide a code snippet to describe what you want to achieve?

[mochacafe]

Hi @shunjd , when we want create a SM training step using Training Step we need to create 'model_estimator' using 'sagemaker.estimator.Estimator' with training ECR image name/ARN , instance_type, output_path.

Is there a way to decide the training ECR image name/ARN and output_path based during Step function initiation and add them as parameters to execution input?

Eg: execution_input ={date:<>,training_ECR_image:<>,training_ouput_path<>}