论坛
BBS
会员中心
Space
用户
登录
用户名
Email
自动登录
找回密码
密码
登录
立即注册
搜索
搜索
本版
帖子
用户
本版
帖子
用户
CUDA
技术
工具
资源
入门
社区
论坛
开启辅助访问
切换到宽版
好友
帖子
收藏
道具
勋章
任务
设置
我的收藏
退出
腾讯QQ
微信登录
论坛
›
社区信息
›
GPU Hackathon
›
TRT Hackathon
›
查看内容
发表新帖
查看:
6062
|
回复:
0
问:可以将包含可变维度的input的pytorch 模型,导出成onnx吗...
GPUS-Cat
GPUS-Cat
当前离线
GPUS-Cat
管理员
|
显示全部楼层
|
阅读模式
1
#
电梯直达
发表于 2021-4-14 12:40:56
6062
0
问:可以将包含可变维度的input的pytorch 模型,导出成onnx吗?比如NLP模型,对可变长度字符串的输入。
答:可以,导出的时候你需要把对应的维度设置dynamic_axes, 请参考:
https://pytorch.org/tutorials/ad ... th_onnxruntime.html
使用道具
举报
回复
高级模式
B
Color
Image
Link
Quote
Code
Smilies
您需要登录后才可以回帖
登录
|
立即注册
发表回复