Can 2D segmentation function be added to UNetR?

[lbwdruid]

Thanks for the wonderful work of UNetR. I was wondering if 2D segmentation function could be added to UNetR? Right now, based on my understanding of the codes, the UNetR only works with 3D datasets. I'm working on 2D ultrasound segmentation, so it would be great if 2D segmentation could work with UNetR.

[wyli]

I think you can set spatial_dims=2

MONAI/monai/networks/nets/unetr.py

Line 42 in fdb16ac

spatial_dims: int = 3,

(converting this to a discussion.)