bert_squad_v1_sim1.sh · littlezheng/onnx_convert - Gitee.com

forked from ahqzy/onnx_convert

加入 Gitee

与超过 1200万开发者一起发现、参与优秀开源项目，私有仓库也完全免费：）

克隆/下载

bert_squad_v1_sim1.sh 283 Bytes

一键复制编辑原始数据按行查看历史

提交于 2023-03-16 20:12 . add bert sh

python model_convert.py   --model_path  /home/zqiu/models/bert_squad_metax_v1.pb  --model_type tf-graph  --output ./pb.onnx --inputs input_ids:0[1,128],input_mask:0[1,128],segment_ids:0[1,128]  --outputs start_prob:0,end_prob:0  --fuse_layernorm 0 --fuse_gelu 0

python ./dump.py

马建仓 AI 助手

尝试更多

代码解读

代码找茬

代码优化

Python

1

https://gitee.com/littezheng/onnx_convert.git

git@gitee.com:littezheng/onnx_convert.git

littezheng

onnx_convert

onnx_convert

master