model = resnext50_32x4d()
total = sum([param.nelement() for param in model.parameters()])
print("total param: {:.2f}M".format(total/1e6)) #总共的参数数量
统计Pytorch模型参数
最新推荐文章于 2024-03-29 17:55:31 发布
model = resnext50_32x4d()
total = sum([param.nelement() for param in model.parameters()])
print("total param: {:.2f}M".format(total/1e6)) #总共的参数数量