用户
 找回密码
 立即注册
发表于 2021-4-14 12:40:56
19920
问:可以将包含可变维度的input的pytorch 模型,导出成onnx吗?比如NLP模型,对可变长度字符串的输入。
答:可以,导出的时候你需要把对应的维度设置dynamic_axes, 请参考:https://pytorch.org/tutorials/ad ... th_onnxruntime.html

使用道具 举报 回复
发新帖
您需要登录后才可以回帖 登录 | 立即注册