torchsummary 打印 Pytorch 模型
2019-01-09 本文已影响0人
深度学习努力中
torchsummary 库实现类似于 Keras 的model.summary()的功能，以便于打印 PyTorch 模型信息.
安装
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就会使用
