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