Skip to content

Repository files navigation

BERT 模型精简化的预测程序

以前做BERT模型时从BERT源码以及其他代码仓库中抽取出来的最精简的BERT预测代码。

包括bert和Albert。

使用方式

代码中带有模型文件,如果直接使用我的模型则不用修改任何内容。 另外就是我记得BERT最早是采用TensorFlow 1.x版本,后来支持了TensorFlow 2.x版本, 在此基础上有日志函数可能有出入,因为过了好久不想搭建环境测试,如果有人使用 过程中遇到问题的话可以提Issues,我可以帮忙处理。

About

BERT simply prediction, BERT最精简预测代码

Resources

Stars

0 stars

Watchers

1 watching

Forks

Releases

Packages

Contributors

Languages