def arguments(self) -> RequestType:
"""The arguments dict that is used to call `create_training_job`.
NOTE: The CreateTrainingJob request is not quite the args list that workflow needs.
The TrainingJobName and ExperimentConfig attributes cannot be included.
"""
self.estimator._prepare_for_training()
train_args = _TrainingJob._get_train_args(
self.estimator, self.inputs, experiment_config=dict()
)
request_dict = self.estimator.sagemaker_session._get_train_request(**train_args)
request_dict.pop("TrainingJobName")
return request_dict
def arguments(self) -> RequestType:
"""The arguments dictionary that is used to call `create_training_job`.
NOTE: The `CreateTrainingJob` request is not quite the args list that workflow needs.
The `TrainingJobName` and `ExperimentConfig` attributes cannot be included.
"""
if self.step_args:
request_dict = self.step_args
else:
self.estimator._prepare_for_training(self.job_name)
train_args = _TrainingJob._get_train_args(self.estimator,
self.inputs,
experiment_config=dict())
request_dict = self.estimator.sagemaker_session._get_train_request(
**train_args)
if "HyperParameters" in request_dict:
request_dict["HyperParameters"].pop("sagemaker_job_name", None)
request_dict.pop("TrainingJobName", None)
request_dict.pop("ExperimentConfig", None)
return request_dict
