Model summary in PyTorch similar to `model.summary()` in Keras

Model summary in PyTorch similar to `model.summary()` in Keras

 Keras has a neat API to view the visualization of the model which is very helpful while debugging your network. Here is a barebone code to try and mimic the same in PyTorch. The aim is to provide information complementary to, what is not provided by print(your_model) in PyTorch.

 

Usage

pip install torchsummary

 

详见https://github.com/sksq96/pytorch-summary

 

记录一个好用的pytorch工具