深度学习网络结构可视化--torchsummary

直接在代码里（在调用模型时）加入下面代码块

 from torchsummary import summary
    summary(model, (3, 28, 28))

ModuleNotFoundError: No module named ‘torchsummary’
没下载该模型，pip一下

pip install torchsummary -i https://pypi.tuna.tsinghua.edu.cn/simple/

运行一下，在训练前，输出各参数