Can't do Segmentation: CUDA error: device-side assert triggered

I had the same problem. Turned out I had saved the modified images in the wrong format with floats

Hi, I’m trying to segmet some images where masks have data between [0,255]; after a lot of searches I tried your code that seems to solve my problem, but when I try do display data:

data.show_batch(2, figsize=(10,7))

I got this error:

BrokenPipeError: [Errno 32] Broken pipe

Any ideas? Thanks for your help.

(fastai version: 1.0.55)

Thanks @imanol this fixed my problem!