使用tensorflow2.15.0版跑bert模型遇到的问题记录

2023-12-04

背景

使用官方的bert模型 https://github.com/google-research/bert >https://github.com/google-research/bert 作文本分类时(运行 run_classifier.py 函数),遇到的一些问题记录

问题记录

  1. 官方模型要求的版本是 tensorflow >= 1.11.0 ,现在安装的是 2.15.0 ,2.x版和1.x版之间有一些函数变了
  2. 训练时,一部分问题参考 https://zhuanlan.zhihu.com/p/390004994 >https://zhuanlan.zhihu.com/p/390004994
  3. 预测时,问题参考 Bert模型预测时出现的问题:ValueError: Assignment map with scope only name bert/embeddings/layer_normalizations >https://blog.csdn.net/m0_69033523/article/details/130224805
  4. 使用自己数据集时,需要新建一个数据类,实现里面的输入文本和标签部分怎么取
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

使用tensorflow2.15.0版跑bert模型遇到的问题记录 的相关文章

随机推荐