update 2d_slices_from_3d_training.ipynb

Signed-off-by: Wenqi Li <[email protected]>

Author: wyli

modules/2d_slices_from_3d_training.ipynb

@@ -501,7 +501,7 @@
     "            device=torch.device(\"cpu\"),\n",
     "            padding_mode=\"replicate\",\n",
     "        )\n",
-    "        val_output = slice_inferer(val_images, model)\n",
+    "        val_output = slice_inferer(val_images, model).to(\"cpu\")\n",
     "        dice_metric(y_pred=val_output > 0.5, y=val_data[\"seg\"])\n",
     "        print(\"Dice: \", dice_metric.get_buffer()[-1][0])\n",
     "        fig = plt.figure(figsize=(10, 4))\n",
@@ -549,7 +549,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.8.12"
+   "version": "3.8.0"
   }
  },
  "nbformat": 4,