TensorFlow pipemode example

Author: owen-t

sagemaker-python-sdk/tensorflow_pipemode_example/tensorflow_pipemode_example.ipynb

@@ -136,9 +136,7 @@
     "                        output_path=model_artifacts_location,\n",
     "                        code_location=custom_code_upload_location,\n",
     "                        train_instance_count=1,\n",
-    "                        # TODO: Remove once sagemaker-tensorflow-extensions is installed in our images\n",
-    "                        image_name='637548144040.dkr.ecr.us-west-2.amazonaws.com/tf-pipemode:1.7.0-cpu-py2-fwmode',\n",
-    "                        training_steps=10000,\n",
+    "                        training_steps=1000,\n",
     "                        evaluation_steps=100,\n",
     "                        train_instance_type='ml.c4.xlarge')"
    ]
@@ -162,8 +160,8 @@
     "# use the region-specific sample data bucket\n",
     "region = boto3.Session().region_name\n",
     "\n",
-    "train_data = 's3://sagemaker-sample-data-{}/tensorflow/pipemode/train'.format(region)\n",
-    "eval_data = 's3://sagemaker-sample-data-{}/tensorflow/pipemode/eval'.format(region)\n",
+    "train_data = 's3://sagemaker-sample-data-{}/tensorflow/pipe-mode/train'.format(region)\n",
+    "eval_data = 's3://sagemaker-sample-data-{}/tensorflow/pipe-mode/eval'.format(region)\n",
     "\n",
     "tensorflow.fit({'train':train_data, 'eval':eval_data})\n"
    ]