pytorch和tensorflow计算模型参数量
pytorchtotal = sum([param.nelement() for param in model.parameters()])
print("Number of parameters: ",total)
parm=np.sum([np.prod(v.get_shape().as_list()) for v in tf.trainable_variables()])
print(‘Number of parameters:’,parm)
