1 Star 2 Fork 0

NLP小学生 / bert-classification-inference-trt

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
贡献代码
同步代码
取消
提示: 由于 Git 不支持空文件夾,创建文件夹后会生成空的 .keep 文件
Loading...
README

基于TensorRT的BERT分类模型推理

本代码库与bert-classification-train-tf2(简称bert-train)搭配使用,bert-train负责训练BERT分类模型,本代码库负责推理。

步骤

  1. 用bert-train训练模型,得到SavedModel文件夹,命名为saved_model/
  2. build_engine/目录将saved_model/转换为TensorRT Engine文件,记为my_model.engine。具体步骤见build_engine/README.md
  3. 基于my_model.engine,用inference/目录进行推理。具体步骤见inference/README.md

空文件

简介

取消

发行版

暂无发行版

贡献者

全部

近期动态

加载更多
不能加载更多了
1
https://gitee.com/nlp_pupil/bert-classification-inference-trt.git
git@gitee.com:nlp_pupil/bert-classification-inference-trt.git
nlp_pupil
bert-classification-inference-trt
bert-classification-inference-trt
master

搜索帮助