Pytorch 如何了解型号';输入大小是否为onnx?

Pytorch 如何了解型号';输入大小是否为onnx?,pytorch,onnx,Pytorch,Onnx,pytorch中未指定输入的大小。只需调整内核的大小即可生成输出。WinMLDashboard显示图像输入的宽度和高度。这怎么可能呢?你是说当你将网络从pytorch序列化到onnx时?因为从pytorch导出时,需要根据 谢谢你,我为什么不考虑呢?我真傻。您有没有用C++来运行ONNX的好的参考?如果您有NVIDIA硬件,最好的办法是使用TunSRRT加载ONNX网络。这将为您提供最快的输出,但是onnx有一个运行时引擎:,您可以从C使用它来运行您的推断。 dummy_input = torc

pytorch中未指定输入的大小。只需调整内核的大小即可生成输出。WinMLDashboard显示图像输入的宽度和高度。这怎么可能呢?

你是说当你将网络从pytorch序列化到onnx时?因为从pytorch导出时,需要根据


谢谢你,我为什么不考虑呢?我真傻。您有没有用C++来运行ONNX的好的参考?如果您有NVIDIA硬件,最好的办法是使用TunSRRT加载ONNX网络。这将为您提供最快的输出,但是onnx有一个运行时引擎:,您可以从C使用它来运行您的推断。
dummy_input = torch.randn(10, 3, 224, 224, device='cuda')
model = torchvision.models.alexnet(pretrained=True).cuda()
input_names = [ "actual_input_1" ] 
output_names = [ "output1" ]

torch.onnx.export(model, dummy_input, "alexnet.onnx", verbose=True, 
input_names=input_names, output_names=output_names)