编程知识 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://cdmana.com/2021/11/20211124170241225i.html

Scroll to Top