2021
07-22
07-22
pytorch如何获得模型的计算量和参数量
方法1自带pytorch自带方法,计算模型参数总量total=sum([param.nelement()forparaminmodel.parameters()])print("Numberofparameter:%.2fM"%(total/1e6))或者total=sum(p.numel()forpinmodel.parameters())print("Totalparams:%.2fM"%(total/1e6))方法2编写代码计算模型参数总量和模型计算量defcount_params(model,input_size=224):#param_sum=0withopen('models.txt','w')asfm:f...
继续阅读 >