torch summary - Serbipunk/notes GitHub Wiki

https://github.com/tyleryep/torch-summary

Torch-summary provides information complementary to what is provided by print(your_model) in PyTorch,

torch-summary提供补充的信息，通过print(your_model)

similar to Tensorflow's model.summary() API to view the visualization of the model, which is helpful while debugging your network.

这点和tensorflow API中model.summary()可视化模型类似，有助于网络的调试。

In this project, we implement a similar functionality in PyTorch and create a clean, simple interface to use in your projects.

这个项目中，我们实现了类似的功能，建立了简单清爽的接口供你调用。