6
运行时
migraphx框架是否支持Vit模型
2025-12-05 09:37:49
76
模型转换成migraphx框架mxr模型后, 原始输入的shape: pixel\_values torch.Size([1, 3, 224, 224]) input\_ids torch.Size([1, 10]) attention\_mask torch.Size([1, 10]) migraphx模型的shape: 'pixel\_values': {1, 3, 224, 224}, {150528, 50176, 224, 1} 'input\_ids': {1, 1}, {1, 1}, 'attention\_mask': {1, 1}, {1, 1} 导致最后推理结果错误。 目前migraphx是否支持,有没有使用示例参考
全部评论(1)
只看楼主
L
lim
1楼
麻烦参考这里 https://www.sourcefind.cn/#/model-zoo/1712380044301778946
2025/12/5
没有更多内容了
--
楼主
Level.3
0 帖子
楼主热帖
暂无数据