小编sta*_*kos的帖子

Hugging Face 的 Transformers 库中的 Trainer 使用的损失函数是什么?

Hugging Face 的 Transformers 库中的 Trainer 使用的损失函数是什么?

我正在尝试使用Hugging Face 的 Transformers 库中的Trainer 类来微调 BERT 模型。

在他们的文档中,他们提到可以通过重写compute_loss类中的方法来指定自定义损失函数。但是,如果我不进行方法覆盖并使用 Trainer 直接微调 BERT 模型以进行情感分类,那么使用的默认损失函数是什么?它是分类交叉熵吗?谢谢!

python nlp artificial-intelligence machine-learning huggingface-transformers

11
推荐指数
1
解决办法
2万
查看次数