KeyError: ‘roformer‘错误解决

最新推荐文章于 2024-08-25 22:10:23 发布

IT_zhouxy

最新推荐文章于 2024-08-25 22:10:23 发布

阅读量714

点赞数

分类专栏：自然语言处理文章标签： python bug

本文链接：https://blog.csdn.net/IT_zhouxy/article/details/118692877

版权

自然语言处理专栏收录该内容

3 篇文章 0 订阅

订阅专栏

在尝试使用Roformer模型时遇到KeyError，问题源于bert4keras模块版本不匹配。通过在终端运行`pip install bert4keras==0.10.4`更新模块后，成功解决了问题。

摘要由CSDN通过智能技术生成

近日，在使用苏神的roformer模型时提示：KeyError: 'roformer错误。详情如下所示

Traceback (most recent call last):
  File "G:/roformer-main/train.py", line 129, in <module>
    return_keras_model=False
  File "C:\Users\zhou\Anaconda3\envs\extract_text\lib\site-packages\bert4keras\models.py", line 2274, in build_transformer_model
    MODEL = models[model]
KeyError: 'roformer'