Closed mahmoodn closed 2 years ago
@nv-jinhosuh Any idea about this?
@mahmoodn For 3D-UNet, due to samples being in non-uniform sizes/shapes, batchsize == 1 is only supported. There is a possible improvement in supporting batchsize > 1 for 'sliding window inference', but this is not supported for v2.0 codebase.
Thanks a lot. I got it.
Using RTX 3080 and a batch size of 2 or 4 for 3d-unet, I get these messages
The run continues, so I am not sure if such messages are important. If I set the batch size to 1, there is no such memory message. But again I am not sure if that is a correct setting. Any thoughts on that?