net = Net()
dummy_input = torch.randn(1, 3, 224, 224, device='cuda')
torch.onnx.export(net, (dummy_input), "1.onnx", verbose=True, input_names=['input1'], output_names=['output1'],opset_version=11)
pytorch导出onnx
最新推荐文章于 2024-05-14 20:58:16 发布