无法从transformers库导入TrainingArguments、Trainer的问题咨询
问题根因
你安装的transformers 2.1.1版本过旧,TrainingArguments和Trainer是transformers 3.1.0版本之后才新增的封装训练工具类,2.x版本未内置这两个API,因此触发导入报错。
解决方案
方案1:升级transformers版本(优先推荐)
你当前环境的PyTorch 1.7.1最高兼容transformers 4.17.0版本,直接执行以下命令升级即可:
pip install transformers==4.17.0
升级后原有导入代码可以直接正常运行,和你已安装的TensorFlow 2.2.0、PyTorch 1.7.1不存在兼容性冲突,无需调整其他依赖。
方案2:保留旧版本的替代实现
如果有其他依赖限制无法升级transformers,就放弃使用封装的Trainer工具,手动实现训练逻辑:
- 调整导入语句,只保留旧版本支持的类:
from transformers import AutoTokenizer, AutoConfig, AutoModelForSequenceClassification
- 后续按照PyTorch自定义训练循环的标准流程完成BERT微调即可。
内容的提问来源于stack exchange,提问作者jacqui_suis
相关产品推荐
相关产品推荐

