torchsummary 打印 Pytorch 模型

torchsummary 库实现类似于 Keras 的model.summary()的功能，以便于打印 PyTorch 模型信息.

Github - pytorch-summary

安装

pip install torchsummary

或

（推荐）git clone https://github.com/sksq96/pytorch-summary

使用方法

from torchsummary import summary

summary(pytorch_model, input_size=(channels, H, W))

其中，input_size是必须设定的，以对网络进行一次 forward.

下载后看readme就会使用