lists and Variables supported as JIT inputs/outputs. Dictionaries and strings are also accepted but

 

Only tuples, lists and Variables supported as JIT inputs/outputs. Dictionaries and strings are also accepted but their usage is not recommended. But got unsupported type builtin_function_or_method

 

报错原因：input没有提供，应该是torch的tensor：

input = torch.randn(1, 3, 48, 224)  # .cuda()

 torch.onnx.export(model, input, onnx_name, verbose=False)