pytorch - RuntimeError：仅支持作为 JIT 输入的元组、列表和变量，但得到了 NoneType

Question

我的代码是

a=torch.randn(1,80,100,requires_grad=True)
torch.onnx.export(waveglow,a, "waveglow.onnx")

我正在尝试将 PyTorch 模型导出为 ONNX 格式，以便可以在 TensorRT 中使用它。在 PyTorch 中测试我的模型时，输入张量维度为 (1,80,x)，其中 x 取决于输入文本长度（我使用的模型是名为 waveglow 的 TTS 模型）

当我尝试运行上述代码以将模型导出到 onnx 时，我总是收到此错误

RuntimeError: Only tuples, lists and Variables supported as JIT inputs, but got NoneType

请帮忙

Answer 1

鉴于您有 NoneType，也许您可​​以检查是否有实际输入，因为事实是，您实际上得到了None.

另外，有什么理由不使用变量？变量将您的输入转换为可以作为输入接受的张量torch.onnx.export。