从transformers库导入TFTrainer时出错

Question

我一直在跟着一个教程，学习如何调整huggingface模型的参数。不过，我无法导入TFTrainer这个模块。请问这个模块是不是已经被弃用了，还是说我遇到了其他问题？

导入语句

from transformers import TFDistilBertForSequenceClassification, TFTrainingArguments, TFTrainer

错误信息

ImportError: cannot import name 'TFTrainer' from 'transformers'

Answer 1

不要使用 TFTrainer，改用 Trainer。

可以使用下面的代码：

from transformers import TFDistilBertForSequenceClassification, TFTrainingArguments, Trainer