编程知识 cdmana.com

transformers库中的Trainer API中没有设计得到train acc吗?请教怎么得到train acc

使用huggingface 的transformers库进行预训练模型的训练时,发现只能得到train loss。
而且compute_metrics好像只能计算验证的acc和loss。




参考答案1:

如果你需要train acc,那么在Trainer中将eval_dataset指定成train_data,也就是让验证集变成训练集就是了,然后compute_metrics一样,就算的就是训练集的了出来的就是train acc和train loss,但是这样没啥意义




参考答案2:

版权声明
本文为[CSDN问答]所创,转载请带上原文链接,感谢
https://ask.csdn.net/questions/7578157

Scroll to Top