python 計算模型參數總量

計算模型參數總量 方法1：pytorch自帶方法，計算模型參數總量 total = sum([param.nelement() for param in model.parameters()]) print(「Number of parameter: %.2fM」 % (total/1e6)) 例子： 紅色斷點放到下一行，並用debug運行。結果爲： 方法2：用工具包thop 要安裝：pip in

>>阅读原文<<