5
框架组件
MIGraphx多输出
2024-03-19 09:30:55
67

我想使用MIGraphx对多个模型进行加速，模型有多个输出，怎么区分MIGraphx的多个输出和onnx模型的输出的对应关系呢，能通过输出节点名字进行关联吗。

全部评论（4）
只看楼主
网
网友OX7fPe
楼主
2楼
感谢解答，请问能否做成自动化的，训练好后的模型不需要人工参与，只需要脚本就能转成MIGraphX。
2023/8/14
Q
qianqing
回复 @网友OX7fPe
目前MIGraphX主要支持ONNX格式的模型，如果是pytorch训练的模型，通过torch.onnx.export可以直接将pytorch模型转换成onnx格式的模型，如果是tensorflow训练的模型，可以通过tf2onnx转换成onnx格式的模型
2023/8/15
Q
qianqing
回复 @网友OX7fPe
可以直接使用driver工具将onnx模型转换为MXR格式的模型，详情参考《MIGraphX教程》模型序列化章节
2024/3/19
Q
qianqing
1楼
通过netron查看网络结构，MIGraphX的输出节点的顺序跟netron查看到的顺序是一致的
2023/8/14
没有更多内容了
--
楼主
Level.1
0 帖子
楼主热帖
暂无数据
常见问题新帖
暂无数据