Fixed malformed TensorFlow estimator declaration. (#1628)

* Fixed malformed TensorFlow estimator declaration. * Removed extraneous output. Co-authored-by: Eric Johnson <[email protected]>
aws · Nov 11, 2020 · d483cac · d483cac
1 parent a688ce3
commit d483cac
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/aws_sagemaker_studio/frameworks/tensorflow_mnist/tensorflow_mnist.ipynb b/aws_sagemaker_studio/frameworks/tensorflow_mnist/tensorflow_mnist.ipynb
@@ -137,11 +137,10 @@
     "\n",
     "estimator = TensorFlow(entry_point=training_script,\n",
     "                       role=role,\n",
-    "                       train_instance_count=1,\n",
-    "                       train_instance_type='ml.p2.xlarge',\n",
+    "                       instance_count=1,\n",
+    "                       instance_type='ml.p2.xlarge',\n",
     "                       framework_version=tf_version,\n",
-    "                       py_version='py3',\n",
-    "                       distributions={'parameter_server': {'enabled': True}})"
+    "                       py_version='py3')"
    ]
   },
   {
@@ -259,14 +258,21 @@
    "source": [
     "predictor.delete_endpoint()"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": []
   }
  ],
  "metadata": {
   "instance_type": "ml.t3.medium",
   "kernelspec": {
    "display_name": "Python 3 (Data Science)",
    "language": "python",
-   "name": "python3__SAGEMAKER_INTERNAL__arn:aws:sagemaker:us-east-2:429704687514:image/datascience-1.0"
+   "name": "python3__SAGEMAKER_INTERNAL__arn:aws:sagemaker:us-east-1:081325390199:image/datascience-1.0"
   },
   "language_info": {
    "codemirror_mode": {