python之路

python-基础-计算模型参数总量

计算模型参数总量方法1：pytorch自带方法，计算模型参数总量total = sum([param.nelement() for param in model.parameters()])print(“Number of parameter: %.2fM” % (total/1e6))例子：...
2023-07-13 17:01 阅读数 39
技术
下载桌面版
GitHub
百度网盘（提取码：draw）
Gitee
云服务器优惠
阿里云优惠券
腾讯云优惠券
华为云优惠券
站点信息
问题反馈
邮箱：[email protected]
QQ群：766591547
关注微信
©2020-2024 ioDraw All rights reserved, Privacy Policy