Patch resolution of UNETR

[riku-ops]

By default, the patch resolution is defined as 16×16×16. Due to GPU memory limitations, if we change it to 32×32×32, how should the model architecture be modified?

[id-b3]

Hi, the patch size for the UNETR is hard-coded to 16 due to the need for modifying the upsampling blocks if the patch size is different.
This has been covered in a previous discussion, in essense you will need to change the patch size in the model and then add an nn.Upsample block to the model.

#4461