minor fixes to notebooks

NVIDIA-AI-IOT · Jan 16, 2020 · cc1112a · cc1112a
1 parent 9a4bfab
commit cc1112a
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 8 deletions.
diff --git a/notebooks/collision_avoidance/live_demo.ipynb b/notebooks/collision_avoidance/live_demo.ipynb
@@ -237,7 +237,12 @@
    },
    "outputs": [],
    "source": [
+    "import time\n",
+    "\n",
     "camera.unobserve(update, names='value')\n",
+    "\n",
+    "time.sleep(0.1)  # add a small sleep to make sure frames have finished processing\n",
+    "\n",
     "robot.stop()"
    ]
   },
@@ -302,9 +307,9 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.6.7"
+   "version": "3.6.9"
   }
  },
  "nbformat": 4,
- "nbformat_minor": 2
+ "nbformat_minor": 4
 }
diff --git a/notebooks/road_following/live_demo.ipynb b/notebooks/road_following/live_demo.ipynb
@@ -317,7 +317,12 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "import time\n",
+    "\n",
     "camera.unobserve(execute, names='value')\n",
+    "\n",
+    "time.sleep(0.1)  # add a small sleep to make sure frames have finished processing\n",
+    "\n",
     "robot.stop()"
    ]
   },
@@ -348,9 +353,9 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.6.7"
+   "version": "3.6.9"
   }
  },
  "nbformat": 4,
- "nbformat_minor": 2
+ "nbformat_minor": 4
 }
diff --git a/notebooks/road_following/train_model.ipynb b/notebooks/road_following/train_model.ipynb
@@ -246,7 +246,7 @@
     "        optimizer.zero_grad()\n",
     "        outputs = model(images)\n",
     "        loss = F.mse_loss(outputs, labels)\n",
-    "        train_loss += loss\n",
+    "        train_loss += float(loss)\n",
     "        loss.backward()\n",
     "        optimizer.step()\n",
     "    train_loss /= len(train_loader)\n",
@@ -258,7 +258,7 @@
     "        labels = labels.to(device)\n",
     "        outputs = model(images)\n",
     "        loss = F.mse_loss(outputs, labels)\n",
-    "        test_loss += loss\n",
+    "        test_loss += float(loss)\n",
     "    test_loss /= len(test_loader)\n",
     "    \n",
     "    print('%f, %f' % (train_loss, test_loss))\n",
@@ -293,9 +293,9 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.6.7"
+   "version": "3.6.9"
   }
  },
  "nbformat": 4,
- "nbformat_minor": 2
+ "nbformat_minor": 4
 }