ValueError: 'labels.shape' 必须等于 'logits.shape',最后一个维度除外。收到: labels.shape=(4, 72) 和 logits.shape=(4, 70, 50257)

ValueError: `labels.shape` must equal `logits.shape` except for the last dimension. Received: labels.shape=(4, 72) and logits.shape=(4, 70, 50257)

提问人:StormVisionary 提问时间:10/24/2023 更新时间:10/24/2023 访问量:16

问:

尝试使用具有 2 列标签(正式和非正式)的 excel 数据表微调 gpt-2 模型时,不断出现此错误。正在制作模型以将正式给定的句子转换为非正式的句子。

训练部分代码如下:

我也尝试了没有稀疏的分类交叉熵。似乎没有多大帮助。

机器学习 NLP 训练-数据 微调

评论


答: 暂无答案