Pytorch中获取模型摘要的3种方法

在pytorch中获取模型的可训练和不可训练的参数,层名称,内核大小和数量。

Pytorch nn.Module 类中没有提供像与Keras那样的可以计算模型中可训练和不可训练的参数的数量并显示模型摘要的方法 。所以在这篇文章中,我将总结我知道三种方法来计算Pytorch模型中可训练和不可训练的参数的数量。

直接手写代码

最直接的办法就是我们自己手写代码代码实现这个功能,所以这里我自己实现了一个函数,函数中为了漂亮所以引入了PrettyTable的包

 

完整文章:

https://avoid.overfit.cn/post/bfed756d1d5147a89f6d8d911b6d29dd

posted @ 2022-08-27 10:08  deephub  阅读(30)  评论(0编辑  收藏  举报