Pytorch中，变量参数，用numel得到参数数目，累加
def get_parameter_number(net):
total_num = sum(p.numel() for p in net.parameters())
trainable_num = sum(p.numel() for p in net.parameters() if p.requires_grad)
return {'Total': total_num, 'Trainable': trainable_num}
Keras中，直接使用model的summary函数
model = k_model() model.summary()
以上这篇PyTorch和Keras计算模型参数的例子就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持本站。
![[Pytorch框架] 4.2.1 使用Visdom在 PyTorch 中进行可视化](https://www.kunjuke.com/file/css/thumb/2.jpg/280-180.jpg)
![[Pytorch框架] 5.1 kaggle介绍](https://www.kunjuke.com/file/css/thumb/8.jpg/280-180.jpg)
