Pytorch 统计模型参数量 param.numel()

放肆的年华 提交于 2020-03-04 03:34:49

Pytorch 统计模型参数量

param.numel()

返回param中元素的数量
在这里插入图片描述

统计模型参数量

num_params = sum(param.numel() for param in net.parameters())
print(num_params)
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!