昇腾社区首页
支持
消息
我的论坛主页
论坛首页
/
计算模型参数（params）、浮点运算次数（FLOPs）、内存占用（memory）和运存占用（MemR+W）
收藏回复举报
计算模型参数（params）、浮点运算次数（FLOPs）、内存占用（memory）和运存占用（MemR+W）
t('forum.solved') 已解决
发表于2024-06-20 11:09:35
0 查看
请问有没有类似torchstat这样的库或API来获取模型的模型参数（params）、浮点运算次数（FLOPs）、内存占用（memory）和运存占用（MemR+W）等信息

我要发帖子

我们使用cookie来确保您的高速浏览体验。继续浏览本站，即表示您同意我们使用cookie。查看详情