Skip to content

Commit 7cd161a

Browse files
author
Christian Osendorfer
committed
Hyperparameters are json encoded at the end of setting up training config.
1 parent 3095949 commit 7cd161a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/sagemaker/estimator.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -686,7 +686,7 @@ def _script_mode_hyperparam_update(self, code_dir: str, script: str) -> None:
686686
hyperparams[JOB_NAME_PARAM_NAME] = self._current_job_name
687687
hyperparams[SAGEMAKER_REGION_PARAM_NAME] = self.sagemaker_session.boto_region_name
688688

689-
self._hyperparameters.update(EstimatorBase._json_encode_hyperparameters(hyperparams))
689+
self._hyperparameters.update(hyperparams)
690690

691691
def _stage_user_code_in_s3(self) -> str:
692692
"""Uploads the user training script to S3 and returns the S3 URI.

0 commit comments

Comments
 (0)