update 2d_slices_from_3d_sampling.ipynb

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

Author: wyli

modules/2d_slices_from_3d_sampling.ipynb

@@ -519,7 +519,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).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",