I am using Fastai for Image Segmentation on the Kaggle Dataset - TGS Salt Identification Challenge. I have tried the code on both CPU and GPU using colab.
- Link to notebook file using CPU: bit.ly/2I2ZOMX_TGS_CPU
- Link to notebook file using GPU: bit.ly/2Vq0ROZ_TGS_GPU
In CPU notebook, I am getting the following error: Target 255 is out of bounds
I have set div = True in open_mask and have also set num_workers = 0
In GPU notebook I am getting the following error: CUDA error: device-side assert triggered
I have taken the custom metric of IOU. Fastai version: 1.0.60
Kindly have a look and suggest me how to fix the issue.