bert_in_a_flask:dockerized烧瓶API使用TensorFlow 2.0提供ALBERT和BERT预测 源码

BERT&ALBERT放在烧瓶中 此回购演示了如何通过简单的dockerized flask API来提供香草BERT和ALBERT预测。 回购包含和实现,以实现多类分类。 模型实现基于 。 该代码被编写为与和兼容。 如果要使用GPU进行训练,则还需要安装nvidia-container-toolkit 。 有关如何安装它的说明,请参见。 该项目使用进行演示。 新闻 2020年4月2日-现在支持 。 数据和模型加载已得到简化和改进。 2020年6月1日-代码库已全面并移植到 。 现在,该实现基于 。 25-10-2019-错误修正。 代码现在可以运行,并带有数据示例。 代码库仍在构建中,