Merge pull request aws#168 from awslabs/mxnet-nb

Update mxnet.ipynb

Author: leleamol

examples/mxnet/sagemaker-notebooks/mxnet.ipynb

@@ -81,14 +81,11 @@
     "\n",
     "### Instantiate and initialize tornasole hook\n",
     "\n",
-    "**NOTE: In order to enable Tornasole functionality while running the script in SageMaker, the hook must be initialized with 'out_dir = /opt/ml/output/tensors'.**\n",
-    "\n",
     "```\n",
     "    # Create SaveConfig that instructs engine to log graph tensors every 10 steps.\n",
     "    save_config = SaveConfig(save_interval=10)\n",
     "    # Create a hook that logs tensors of weights, biases and gradients while training the model.\n",
-    "    tornasole_path = '/opt/ml/output/tensors'\n",
-    "    hook = TornasoleHook(out_dir=output_s3_uri, save_config=save_config)\n",
+    "    hook = TornasoleHook(save_config=save_config)\n",
     "```\n",
     "\n",
     "### Register Tornasole hook to the model before starting of the training.\n",
@@ -105,15 +102,13 @@
     "```\n",
     "\n",
     "#### Set the mode\n",
-    "Set the mode you are running the job in. This helps you group steps by mode, \n",
-    "for easier analysis. \n",
-    "If you do not specify this, it saves steps under a `default` mode.\n",
-    "\n",
+    "Tornasole has the concept of modes (TRAIN, EVAL, PREDICT) to separate out different modes of the jobs.\n",
+    "Set the mode you are running in your job. Every time the mode changes in your job, please set the current mode. This helps you group steps by mode, for easier analysis. Setting the mode is optional but recommended. If you do not specify this, Tornasole saves all steps under a `GLOBAL` mode. \n",
     "```\n",
     "hook.set_mode(ts.modes.TRAIN)\n",
     "```\n",
     "\n",
-    "Refer [DeveloperGuide_MXNet.md](../DeveloperGuide_MXNet.md) for more details on the APIs Tornasole provides to help you save tensors.\n",
+    "Refer [DeveloperGuide_MXNet.md](../../DeveloperGuide_MXNet.md) for more details on the APIs Tornasole provides to help you save tensors.\n",
     "\n",
     "\n",
     "## SageMaker with Tornasole\n",
@@ -250,7 +245,6 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "import boto3\n",
     "import sagemaker\n",
     "from sagemaker.mxnet import MXNet\n",
     "\n",
@@ -270,7 +264,7 @@
     "\n",
     "The 'entry_point_script' points to the MXNet training script that has the TornasoleHook integrated.\n",
     "\n",
-    "The 'hyperparameters' are the parameters that will be passed to the training script. Please note that the **tornasole_path** parameter is set to be **/opt/ml/output/tensors**. This is **mandatory** when running the training script with SageMaker and Tornasole.\n",
+    "The 'hyperparameters' are the parameters that will be passed to the training script.\n",
     "\n"
    ]
   },
@@ -281,7 +275,7 @@
    "outputs": [],
    "source": [
     "entry_point_script = '../scripts/mnist_gluon_basic_hook_demo.py'\n",
-    "hyperparameters = {'tornasole_path' : '/opt/ml/output/tensors', 'random_seed' : True,  'num_steps': 6}"
+    "hyperparameters = {'random_seed' : True,  'num_steps': 6}"
    ]
   },
   {
@@ -377,7 +371,7 @@
    "outputs": [],
    "source": [
     "entry_point_script = '../scripts/mnist_gluon_vg_demo.py'\n",
-    "bad_hyperparameters = {'tornasole_path' : '/opt/ml/output/tensors', 'random_seed' : True,  'num_steps': 33, 'tornasole_frequency' : 30}"
+    "bad_hyperparameters = {'random_seed' : True,  'num_steps': 33, 'tornasole_frequency' : 30}"
    ]
   },
   {