换成自己的数据集后，出现CUDA ERROR，请教各位大佬这是什么情况

[dujiajianbing]

[xjDUAN184]

我也有这种情况

[Anndra007]

我更换bert的配置文件后就可以了

[dujiajianbing]

我更换bert的配置文件后就可以了 要更换什么呢？

[Anndra007]

我已收到您的来信，敬请等待。

[Anndra007]

https://huggingface.co/bert-base-uncased/tree/main ，我使用了这里面的三个文件

[blueyeee]

https://huggingface.co/bert-base-uncased/tree/main ，我使用了这里面的三个文件

我下载了你说的三个文件进行替换，但是还是一样报错了

[Anndra007]

我已收到您的来信，敬请等待。

[Augenstern-yzh]

自己的数据集，分类的类别数值要连续，比如0，1，2，3，4，5

[Anndra007]

我已收到您的来信，敬请等待。

[TheNorthOne]

我的也是cuda报错，但类别数值是连续的，1 2 3 4……

[Anndra007]

我已收到您的来信，敬请等待。