可视化网络模型结构

第一种：查看参数量 pip install torchsummary #此段代码之显示Output Shape和Param from torchsummary import summary summary(model.cuda(), input_size=(3, 224, 224)) 224分别为输入高度和宽度，3为通道数，界面如下所示 第2种：查看参数量，这种的最详细，但是也是最新的方法，所以还

>>阅读原文<<