pytorch查看模型大小,pytorch数据预测

用resnet50 来举例子

print("resnet50 have {} paramerters in total".format(sum(x.numel() for x in resnet50.parameters())))

其中numel表示含有多少element，通过此操作可以统计模型的参数量有多少

 

另外，两个是一样的，方便debug看其中到底有什么东西

num_params = 0 for param in netG.parameters(): num_params += param.numel() print(num_params / 1e6)