网址:https://netron.app/
点选择模型,将oonx文件拉到netron界面,即可
输出;
如何将pytorch模型转换为onnx的格式?
在测试(训练好的模型)里输入代码
to_onnx(model, 3, 28, 28, 'output/params.onnx')
其中
def to_onnx(model, c, w, h, onnx_name):dummy_input = torch.randn(1, c, w, h, device='cpu')torch.onnx.export(model, dummy_input, onnx_name, verbose=True)