【解决】transformers库调用tensorflow训练模型时报错

最新推荐文章于 2024-04-23 15:26:33 发布

xaaaab

最新推荐文章于 2024-04-23 15:26:33 发布

阅读量637

点赞数

文章标签： tensorflow 人工智能 python bert 深度学习

本文链接：https://blog.csdn.net/foryourkindness/article/details/131530357

版权

描述：在调用transformers库中的与训练模型时，想使用tensorflow的模型但是会报ImportError，

，如：

model = TFBertForSequenceClassification.from_pretrained("bert-base-chinese")
model = AutoModelForMaskedLM.from_pretrained("bert-base-chinese", from_tf=True)

报错：TFBertForMaskedLM requires the TensorFlow library but it was not found in your environment. Checkout the instructions on the installation page: https://www.tensorflow.org/install and follow the ones the ones that match your environment.

尝试：参照ImportError: TFBertForSequenceClassification requires the TensorFlow library but it was not found_VirgilG72的博客-CSDN博客，尝试根据kaggle上的tensorflow和transformers版本进行匹配，如图，但是依旧报错。并且尝试更换transformer版本从4.5、4.8、4.30都无效。