Update notebooks

training/05.distributed-tensorflow-with-parameter-server/05.distributed-tensorflow-with-parameter-server.ipynb

@@ -41,6 +41,28 @@
         "print(\"SDK version:\", azureml.core.VERSION)"
       ]
     },
+    {
+      "cell_type": "markdown",
+      "metadata": {},
+      "source": [
+        "## Diagnostics\n",
+        "Opt-in diagnostics for better experience, quality, and security of future releases."
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "tags": [
+          "Diagnostics"
+        ]
+      },
+      "outputs": [],
+      "source": [
+        "from azureml.telemetry import set_diagnostics_collection\n",
+        "set_diagnostics_collection(send_diagnostics = True)"
+      ]
+    },
     {
       "cell_type": "markdown",
       "metadata": {},
@@ -262,6 +284,11 @@
     }
   ],
   "metadata": {
+    "authors": [
+      {
+        "name": "minxia"
+      }
+    ],
     "kernelspec": {
       "display_name": "Python 3.6",
       "language": "python",

training/05.distributed-tensorflow-with-parameter-server/tf_mnist_replica.py

@@ -263,7 +263,7 @@ def main(unused_argv):
         print("After %d training step(s), validation cross entropy = %g" %
               (FLAGS.train_steps, val_xent))
         if job_name == "worker" and task_index == 0:
-            run = Run.get_submitted_run()
+            run = Run.get_context()
             run.log("CrossEntropy", val_xent)