From 948b44d3a8e7810823c428a68c117c9e69fb1e32 Mon Sep 17 00:00:00 2001 From: Lauren Yu <6631887+laurenyu@users.noreply.github.com> Date: Thu, 25 Jun 2020 13:03:17 -0700 Subject: [PATCH] change: add TF migration documentation to error message --- src/sagemaker/tensorflow/estimator.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/sagemaker/tensorflow/estimator.py b/src/sagemaker/tensorflow/estimator.py index dca755f757..89ffedf54a 100644 --- a/src/sagemaker/tensorflow/estimator.py +++ b/src/sagemaker/tensorflow/estimator.py @@ -160,12 +160,12 @@ def _validate_args(self, py_version): self.py_version, ) - # TODO: add link to docs to explain how to use legacy mode with v2 msg = ( "TF {} supports only legacy mode. Please supply the image URI directly with " "'image_name={}' and set 'model_dir=False'. If you are using any legacy parameters " "(training_steps, evaluation_steps, checkpoint_path, requirements_file), " - "make sure to pass them directly as hyperparameters instead." + "make sure to pass them directly as hyperparameters instead. For more, see " + "https://sagemaker.readthedocs.io/en/v2.0.0.rc0/frameworks/tensorflow/upgrade_from_legacy.html." ).format(self.framework_version, legacy_image_uri) raise ValueError(msg)