Hi, I think I have the same issue, but what is not normal is that I run my code (a bert classifier for sentiment analysis) with a given dataset it worked but when I change the dataset (bigger than the first), it showed me this error : runtime error : the GPU program failed to execute at /pytorch/aten/src/THC/THCBlas.cu:23
jdjddkkk1200×541 89.8 KB