CUDA error: device-side assert triggered after a certain steps

[shayonhuggingface]

if you got the Index Error you can try investigate the label indexing in your data. The out of bounds problem normally comes from that