两行代码统计模型参数量与FLOPs，这个PyTorch小工具值得一试

点击上方“ 磐创AI ”，选择“置顶公众号” 精品文章，第一时间送达python 转载自：机器之心，未经容许不得二次转载git 你的模型到底有多少参数，每秒的浮点运算到底有多少，这些你都知道吗？近日，GitHub 开源了一个小工具，它能够统计 PyTorch 模型的参数量与每秒浮点运算数（FLOPs）。有了这两种信息，模型大小控制也就更合理了。github 其实模型的参数量好算，但浮点运算数并很差

>>阅读原文<<