Skip to content

Tensorflow estimator script mode not working if we pass HyperParameters as part of TrainingStep #72

Closed
@the-elancier

Description

@the-elancier

Hello,

I was trying to use Tensorflow estimator in script mode and I was trying to pass the hyper-parameters as part of stepfunctions.steps.TrainingStep as follows:
Screenshot 2020-08-04 at 6 22 01 PM

But, the training job failed with the following error
Screenshot 2020-08-04 at 6 11 50 PM

Upon investigation I found out that other hyper-parameters that the estimator attaches on its own were not going (ex - 'sagemaker_program', 'sagemaker_submit_directory' etc.)

For example, the following image is the set of hyper-parameters for the job when we give it as part of TrainingStep:
Screenshot 2020-08-04 at 6 21 42 PM

But, if we give the hyper-parameters as part of the estimator:
Screenshot 2020-08-04 at 6 12 59 PM

The reason why I'd want to give hyper-parameters as part of the TrainingStep is that I'd be able to use stepfunctions.inputs.ExecutionInput in order to provide inputs dynamically that might be useful in the script.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions