From adaf0388709685426fd033f1237f5061ba06bcaf Mon Sep 17 00:00:00 2001 From: Wenqi Li Date: Tue, 15 Nov 2022 08:14:04 +0000 Subject: [PATCH] fixes invertd issue Signed-off-by: Wenqi Li --- 3d_segmentation/brats_segmentation_3d.ipynb | 1 + 3d_segmentation/spleen_segmentation_3d.ipynb | 1 + 2 files changed, 2 insertions(+) diff --git a/3d_segmentation/brats_segmentation_3d.ipynb b/3d_segmentation/brats_segmentation_3d.ipynb index d43c25cd16..805f943aa1 100644 --- a/3d_segmentation/brats_segmentation_3d.ipynb +++ b/3d_segmentation/brats_segmentation_3d.ipynb @@ -814,6 +814,7 @@ " meta_key_postfix=\"meta_dict\",\n", " nearest_interp=False,\n", " to_tensor=True,\n", + " device=\"cpu\",\n", " ),\n", " Activationsd(keys=\"pred\", sigmoid=True),\n", " AsDiscreted(keys=\"pred\", threshold=0.5),\n", diff --git a/3d_segmentation/spleen_segmentation_3d.ipynb b/3d_segmentation/spleen_segmentation_3d.ipynb index 83769e0b88..0816a82aaa 100644 --- a/3d_segmentation/spleen_segmentation_3d.ipynb +++ b/3d_segmentation/spleen_segmentation_3d.ipynb @@ -717,6 +717,7 @@ " meta_key_postfix=\"meta_dict\",\n", " nearest_interp=False,\n", " to_tensor=True,\n", + " device=\"cpu\",\n", " ),\n", " AsDiscreted(keys=\"pred\", argmax=True, to_onehot=2),\n", " AsDiscreted(keys=\"label\", to_onehot=2),\n",