Visual model network connection

Thank you to this blogger , Let me know how to visualize the model network , Can only use a handle Pytorch Drag the model of  Netron in , But no network connection , It turned to onnx, That's all right. , thank ！

def test():
    args = parser.parse_args()

    repvgg_build_func = get_RepVGG_func_by_name(args.arch)

    model = repvgg_build_func(deploy=args.mode=='deploy')

    model.eval()
    # convert to onnx
    print('=====convert to onnx=======')
    dummy_input = torch.randn(1, 3, 224, 224)
    torch.onnx.export(model, dummy_input, 'RepVGG-A1.onnx', export_params=True, verbose=False, input_names=['input0'],
                      output_names=['output0'])
    print('=====convert sucess=======')
    exit()

 

【 Network understanding 】RepVGG: Making VGG-style ConvNets Great Again_songlixiangaibin The blog of -CSDN Blog _repvgg Code