Skip to content

Commit ac9b16c

Browse files
authored
fixes invertd issue (#1042)
Signed-off-by: Wenqi Li <[email protected]> Fixes #1041 ### Description issue introduced by Project-MONAI/MONAI#5512 ### Checks <!--- Put an `x` in all the boxes that apply, and remove the not applicable items --> - [ ] Notebook runs automatically `./runner [-p <regex_pattern>]` Signed-off-by: Wenqi Li <[email protected]>
1 parent f43e771 commit ac9b16c

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

3d_segmentation/brats_segmentation_3d.ipynb

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -814,6 +814,7 @@
814814
" meta_key_postfix=\"meta_dict\",\n",
815815
" nearest_interp=False,\n",
816816
" to_tensor=True,\n",
817+
" device=\"cpu\",\n",
817818
" ),\n",
818819
" Activationsd(keys=\"pred\", sigmoid=True),\n",
819820
" AsDiscreted(keys=\"pred\", threshold=0.5),\n",

3d_segmentation/spleen_segmentation_3d.ipynb

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -717,6 +717,7 @@
717717
" meta_key_postfix=\"meta_dict\",\n",
718718
" nearest_interp=False,\n",
719719
" to_tensor=True,\n",
720+
" device=\"cpu\",\n",
720721
" ),\n",
721722
" AsDiscreted(keys=\"pred\", argmax=True, to_onehot=2),\n",
722723
" AsDiscreted(keys=\"label\", to_onehot=2),\n",

0 commit comments

Comments
 (0)