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

[aws]

Feature Request: Cannot pass output path and ECR Image ARN from 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<>}