首页 > 未分类 > pytorch统计模型参数量
2019
11-15

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)

 

最后编辑:
作者:搬运工
这个作者貌似有点懒,什么都没有留下。