Skip to content

Commit 8212458

Browse files
authored
Fix swinunetrv2 2D bug (#7302)
Fixes # . ### Description A few sentences describing the changes proposed in this pull request. ### Types of changes <!--- Put an `x` in all the boxes that apply, and remove the not applicable items --> - [x] Non-breaking change (fix or new feature that would not break existing functionality). - [ ] Breaking change (fix or new feature that would cause existing functionality to change). - [ ] New tests added to cover the changes. - [ ] Integration tests passed locally by running `./runtests.sh -f -u --net --coverage`. - [ ] Quick tests passed locally by running `./runtests.sh --quick --unittests --disttests`. - [ ] In-line docstrings updated. - [ ] Documentation updated, tested `make html` command in the `docs/` folder. Signed-off-by: heyufan1995 <heyufan1995@gmail.com>
1 parent d5585c3 commit 8212458

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

monai/networks/nets/swin_unetr.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1024,7 +1024,7 @@ def __init__(
10241024
self.layers4.append(layer)
10251025
if self.use_v2:
10261026
layerc = UnetrBasicBlock(
1027-
spatial_dims=3,
1027+
spatial_dims=spatial_dims,
10281028
in_channels=embed_dim * 2**i_layer,
10291029
out_channels=embed_dim * 2**i_layer,
10301030
kernel_size=3,

0 commit comments

Comments
 (0)