onnxruntime 推理失败

[gmx1992]

你好，作者，首先很感谢你们的工作，让我学习了新的公式识别思路。我目前遇到一个问题，具体内容：Non-zero status code returned while running Add node. Name:'Add_3182' Status Message: Add_3182: right operand cannot broadcast on dim 0 LeftShape: {2,3828}, RightShape: {6}。虽然问题显示是add算子的输入导致无法广播，但是我使用pytorch-lighting原来的模型（你们的repo中提供了）推理正常。网上有建议说是ort-gpu的版本问题，但是高版本的onnx和onnxruntime又与你们建议的pytorch-lighting 版本(好像是1.4.9)不兼容。所以，想请问，你们有成功实现onnxruntime-gpu的推理吗？以及对应的框架版本。谢谢！