模型转换成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是否支持,有没有使用示例参考