torchsummary计算模型参数量输出的含义

参见此博客https://blog.csdn.net/csdnxiekai/article/details/110517751
值得注意的是:
这里的Input size是输入的特征的大小
Forward/backward pass size是所有输出的大小
Params size是模型的参数量
Estimated Total Size是前三者之和

posted @ 2024-05-30 19:56  windy_fly  阅读(14)  评论(0编辑  收藏  举报