安装对应的库文件

pip install ptflops

代码（get_model_complexity_info)

import torchvision.models as models
from ptflops import get_model_complexity_info
net = models.vgg16() #可以为自己搭建的模型
flops, params = get_model_complexity_info(model, (3,512,512), as_strings=True, print_per_layer_stat=True)  #(3,512,512)输入图片的尺寸
print("Flops: {}".format(flops))
print("Params: " + params)
# python # 机器学习 # 深度学习
Logo
DAMO开发者矩阵

DAMO开发者矩阵，由阿里巴巴达摩院和中国互联网协会联合发起，致力于探讨最前沿的技术趋势与应用成果，搭建高质量的交流与分享平台，推动技术创新与产业应用链接，围绕“人工智能与新型计算”构建开放共享的开发者生态。

更多推荐

同态加密——看不到数据是怎么计算的呢？

看不到数据是怎么计算的？我们计算的是数据的“加密替身”（密文），但这个替身被赋予了一种特殊的“骨骼结构”或“基因”。当我们按照预先定义好的规则（密文运算）来操纵这些替身时，它们内部结构的变化，恰好对应了原始数据（明文）之间我们想要的实际运算结果。解密过程，就是读取这个最终“结构状态”，并将其翻译回数据域的过程。服务器就像一个在黑暗房间里的机器人，它不知道手里拿的积木（密文）原本是什么形状（明文），

avatar DAMO开发者矩阵
cover

老板设置“狼性模式”后，AI把竞争对手客户都挖来了

avatar DAMO开发者矩阵

蚂蚁灵波开源具身大模型LingBot-VLA，跨本体跨任务泛化能力创新高

与高精度空间感知模型 LingBot-Depth配合，LingBot-VLA 能获得更高质量的深度信息表征，通过“视力”的升级，真正做到“看得更清楚、做的更明白”。在 RoboTwin 2.0 仿真基准（包含50项任务）评测中，面对高强度的环境随机化干扰（如光照、杂物、高度扰动），LingBot-VLA 凭借独特的可学习查询对齐机制，高度融合深度信息，操作成功率比 Pi0.5 提升了 9.92%，

avatar DAMO开发者矩阵