I am trying to train on the unet from torch.hub.  Some of the image-label pairs give me a CUDA error: device-side assert triggered.

Screenshot%20from%202019-12-11%2000-52-03806×341 40.9 KB

How Can I fix this?