文件名称：bert_in_a_flask：dockerized烧瓶API，使用TensorFlow 2.0提供ALBERT和BERT预测
文件大小：16KB
文件格式：ZIP
更新时间：2024-02-25 21:30:32
docker flask tensorflow keras transformer
BERT＆ALBERT放在烧瓶中 此回购演示了如何通过简单的dockerized flask API来提供香草BERT和ALBERT预测。 回购包含和实现，以实现多类分类。 模型实现基于 。 该代码被编写为与和兼容。 如果要使用GPU进行训练，则还需要安装nvidia-container-toolkit 。 有关如何安装它的说明，请参见。 该项目使用进行演示。 新闻 2020年4月2日-现在支持 。 数据和模型加载已得到简化和改进。 2020年6月1日-代码库已全面并移植到 。 现在，该实现基于 。 25-10-2019-错误修正。 代码现在可以运行，并带有数据示例。 代码库仍在构建中，
【文件预览】：
bert_in_a_flask-master
----LICENSE(1KB)
----src()
--------app.py(1KB)
--------train.py(3KB)
--------predict.py(2KB)
--------utils()
----docker()
--------api.Dockerfile(190B)
--------dev.Dockerfile(622B)
----requirements.txt(85B)
----.gitignore(46B)
----Makefile(1007B)
----.dockerignore(13B)
----README.md(7KB)