PyTorch模型转onnx格式

点击查看原文

原文作者:老杨 | 来源:老杨说话的地方

因项目需要,把PyTorch模型转为onnx。

Huggingface的模型无法直接转为onnx(仅有部分支持),如果是huggingface的模型,需要先转成PyTorch,详见文章《Huggingface模型转PyTorch模型》

onnx是一种通用的模型架构,详细介绍可参考官网

转换代

...

相关文章

暂无评论

none
暂无评论...
📌 本站已加入 bkq.net.cn,欢迎广大博友交流!