兩行代碼統計模型參數量與FLOPs，這個PyTorch小工具值得一試

點擊上方「 磐創AI 」，選擇「置頂公衆號」 精品文章，第一時間送達python 轉載自：機器之心，未經容許不得二次轉載git 你的模型到底有多少參數，每秒的浮點運算到底有多少，這些你都知道嗎？近日，GitHub 開源了一個小工具，它能夠統計 PyTorch 模型的參數量與每秒浮點運算數（FLOPs）。有了這兩種信息，模型大小控制也就更合理了。github 其實模型的參數量好算，但浮點運算數並很差

>>阅读原文<<