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