Update tensorflow_multi_worker_mirrored_strategy.ipynb dataset

Author: kandakji

training/distributed_training/tensorflow/multi_worker_mirrored_strategy/tensorflow_multi_worker_mirrored_strategy.ipynb

@@ -62,11 +62,11 @@
    "source": [
     "## Training Data\n",
     "\n",
-    "We will use the MNIST dataset has been already loaded to the public S3 buckets ``sagemaker-sample-data-<REGION>`` under the prefix ``tensorflow/mnist``. There are four ``.npy`` file under this prefix:\n",
-    "* ``train_data.npy``\n",
-    "* ``eval_data.npy``\n",
-    "* ``train_labels.npy``\n",
-    "* ``eval_labels.npy``"
+    "We will use the MNIST dataset has been already loaded to the public S3 buckets ``sagemaker-example-files-prod-<REGION>`` under the prefix ``datasets/image/MNIST``. There are four ``.npy`` file under this prefix:\n",
+    "* ``input_train.npy``\n",
+    "* ``input_test.npy``\n",
+    "* ``input_train_labels.npy``\n",
+    "* ``input_test_labels.npy``"
    ]
   },
   {
@@ -77,7 +77,7 @@
    },
    "outputs": [],
    "source": [
-    "training_data_uri = \"s3://sagemaker-sample-data-{}/tensorflow/mnist\".format(region)"
+    "training_data_uri = \"s3://sagemaker-example-files-prod-{}/datasets/image/MNIST/numpy\".format(region)"
    ]
   },
   {
@@ -127,7 +127,7 @@
    "source": [
     "from sagemaker.tensorflow import TensorFlow\n",
     "\n",
-    "local_mode = True\n",
+    "local_mode = False\n",
     "\n",
     "if local_mode:\n",
     "    instance_type = \"local_gpu\"\n",