可视化模型网络连接

感谢这位博主，让我知道了怎么可视化模型网络，只会用把 Pytorch 的模型拖进 Netron 中，但是没有网络连接，原来转为 onnx，就可以了，感谢！

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()

 

【网络理解】RepVGG: Making VGG-style ConvNets Great Again_songlixiangaibin的博客-CSDN博客_repvgg代码