python 计算模型参数总量

计算模型参数总量 方法1：pytorch自带方法，计算模型参数总量 total = sum([param.nelement() for param in model.parameters()]) print(“Number of parameter: %.2fM” % (total/1e6)) 例子： 红色断点放到下一行，并用debug运行。结果为： 方法2：用工具包thop 要安装：pip in

>>阅读原文<<