Considered all comments.

Author: Ragav Venkatesan

advanced_functionality/mxnet_mnist_byom/mxnet_mnist.ipynb

@@ -133,13 +133,15 @@
    "outputs": [],
    "source": [
     "from mnist import train\n",
-    "model = train(data = data)"
+    "model = train(data = data, num_cpus=0, num_gpus=1)"
    ]
   },
   {
    "cell_type": "markdown",
    "metadata": {},
    "source": [
+    "If you want to run the training on a cpu or if you are on an instance with cpus only, pass appropriate arguments. \n",
+    "\n",
     "## Set up hosting for the model\n",
     "\n",
     "### Export the model from mxnet\n",
@@ -228,6 +230,9 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "import logging\n",
+    "logging.getLogger().setLevel(logging.WARNING)\n",
+    "\n",
     "predictor = sagemaker_model.deploy(initial_instance_count=1,\n",
     "                                          instance_type='ml.c4.xlarge')"
    ]
@@ -304,7 +309,7 @@
    },
    "outputs": [],
    "source": [
-    "# sagemaker.Session().delete_endpoint(predictor.endpoint)"
+    "sagemaker.Session().delete_endpoint(predictor.endpoint)"
    ]
   },
   {