tensorflow构建的ckpt文件转pb转onnx文件，深度学习模型推理时加速，以bert模型为例 tensorflow构建的ckpt文件转pb转onnx文件，深度学习模型推理时加速，以bert模型为例1、环境配置2、在构建模型时对输入、输出指定相应的名字3、训练得到的ckpt文件转换成为pb文件4、通过tf2onnx工具将pb文件转为onnx文件 nlp基础应用 2023-08-23 47 点赞 0 评论 71 浏览
