CUDA error: device-side assert triggered

🤗Transformers
1

When I run the GPU (google colab) I always get RuntimeError: CUDA error: device-side assert triggered, with huggingface. When I convert to CPU, it works fine, any solution?

2

its solved, when I corrected the number of label classes, closing

3

Could you elaborate on what you did when you say “corrected the number of label classes”? As in which at step of the code did you have to change it? What was the incorrect number and what was the correct number?

4

Label classes in my dataset between 77-112. It solved this issue when start by 0 and set between 0-46.

I use this code:

data["label"] = data["label"].apply(lambda x: x-77)