神经网络结构可视化-netron

网址：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)