CUDA error: device-side assert triggered

When doing automation of batches to GPU I get this error, I thought the solution was this

The problem still persist.

According to the solution explained in the link I had to change the number of classes:

The whole script ran, the issue is no longer being mention. I can only imagine that all the GPU were being used.


i am also getting same issue with CIFAR100 dataset. when putted to cpu its running fine.