Target out of bounds for UNET

[muellerzr (Zachary Mueller)]

There’s actually a helpful thread if you look hard enough

Dealing with CUDA Device Assist on Segmentation (and other Segmentation issues), some tips! - A Wiki

There are so many times CUDA device assist errors happen, and this is due to the pixels in your mask not aligning how they should be! (sometimes). For example, if I have a binary mask that is [0,255], it will not work because fastai expects [0,1]. So how do we fix this? And how should we watch for this? Let’s look at a few tips and examples. (this is also a Wiki, if you have run into other segmentation issues that are not obvious fixes and fixed them feel free to post your solutions here) Some …