tensorflow BERT学习记录--使用自己的训练集--run_classifier.py
tensorflow bert github上有tensorflow bert的源代码和预训练模型的下载链接
该仓库里的run_classifier.py 用于fine-tuning以及推理,官方的BERT库里,只需要修改run_classifier.py即可完成对新数据集的fine-tuning和推理, 该文件里的代码主要是以下几块:
训练参数Flags的定义, 没什么好说的, 官方...
原创
2019-01-25 16:53:45 ·
10498 阅读 ·
3 评论