weixin_62291315的博客

原创 transformers.trainer compute_metrics的一些踩坑--参数传递与trainer 在计算metrics的相关设置

然后我发现eval_preds传入的严格意义上来说并不是模型的预测结果，在代码里，传入的eval_preds实际上是模型在前向传播后输出的除了loss,label,past_key_value等相关词条之外的内容，具体不清楚，但是我自己添加的词条都被归为模型预测结果传入compute_metrics中作为eval_preds参数了，此时该参数为一个tuple，包含了这些词条.