RuntimeError: The size of tensor a (631) must match the size of tensor b (512) at non-singleton dime

BUG

RuntimeError: The size of tensor a (631) must match the size of tensor b (512) at non-singleton dimension 1

BUG 异常原因分析

在bert训练过程中部分训练样本长度大于512,所以会有该BUG;

解决思路(点赞关注不迷路…)

过滤下训练语料,将长度过长的数据直接丢弃;数据过滤之后再次训练,问题解决;

你可能感兴趣的:(深度学习,机器学习,人工智能)