TensorRT_BERT4TensorRT.zip

共7个文件

py：4个

md：2个

gitignore：1个

需积分: 5 0 下载量 161 浏览量 2024-08-24 12:59:54 上传评论收藏 15KB ZIP 举报

温馨提示

TensorRT_BERT4TensorRT

资源推荐

资源详情

资源评论

收起资源包目录

TensorRT_BERT4TensorRT.zip （7个子文件）

DataXujing-BERT4TensorRT-807c19a

encode

data_processing_new.py 15KB

bert_base_build.py 12KB

readme.md 462B

bert_tensorrt.py 3KB

tokenization.py 12KB

.gitignore 2KB

README.md 25B

共 7 条

# 使用TensorRT做encode 官方有个fast transformer 这个项目demo/Bert下改的 ## 模型转换成.engine文件，在bert_build文件下做如下修改 1. 删除了squad函数 2. 删除了squad的调用 ## 修改后再改data_processing文件 1. 原来是处理MC数据的，现在改成处理一般句子的 ## 修改BERT_TRT 1. 原来是处理MC数据的，现在改成处理一般句子的 OK，完成以上操作，TensorRT的bert就完成了

评论收藏

内容反馈