pytorch bert RuntimeError: CUDA error: device-side assert triggered

在用BERT进行文本分类的时候,这个错误困扰了我一晚上,百度到的答案都是说标签的问题,但是我用放大镜来仔细观察后确认标签没有问题;之后通过与之前的代码对比发现,原来是设置的最大长度过长的问题;输入的句子的容许长度太长了;我从1000改为100就ok了。
问题千千万,仅供参考,如有雷同,不甚荣幸。

我下载的bert max_position_embeddings限制512
 parser.add_argument("--max_seq_length",
                        default = 100,  #就是这厮
                        type = int,
                        help = "字符串最大长度")

你可能感兴趣的:(笔记)