PyTorch和Keras计算模型参数的例子

Pytorch中，变量参数，用numel得到参数数目，累加defget_parameter_number(net):total_num=sum(p.numel()forpi

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 Keras 模型参数