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計算模型引數的例子就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支援我們。