【踩坑】单GPU微调BERT(SQuAD 1.1)
官方代码地址 https://github.com/google-research/bert
用SQuAD 1.1数据集fine fune时,遇到不少坑,在此分享一下
一、环境
1.报错:AttributeError: module ‘tensorflow._api.v2.train’ has no attribute ‘Optimizer’
问题为tensorflow版本不正确,必须使用1.x版本,我使用的是tensorflow1.15.0
注意:python3.8不支持tensorflow1.15.0,
原创
2021-03-06 12:00:55 ·
974 阅读 ·
1 评论