pytorch可视化

torchinfo提供了详细的信息，包括模块信息(每一层的类型、输出shape和参数量)、模型整体的参数量、模型大小、一次前向或者反向传播需要的内存大小等
值得注意的是：
使用的是colab或者jupyter notebook时，想要实现该方法， summary() 一定是该单元(即 notebook中的cell)的返回值，否则就需要使用 print(summary(…)) 来可视化。