BN
那里,春暖花开
这个作者很懒,什么都没留下…
展开
-
pytorch中Conv2d的FLOPs的计算范例
import torchconv = torch.nn.Conv2d(1,8,(2,3))input = torch.rand(1,1,224,224) # batch,channel,width,heightoutput = conv(input)print(output.shape)bn = torch.nn.BatchNorm2d(8)l = [conv,bn]for module in l: class_name = str(module.__class__.__name__原创 2020-06-13 16:14:06 · 1188 阅读 · 1 评论 -
查看Conv2d,BN两类module的参数大小
import torchconv = torch.nn.Conv2d(1,8,(2,3))bn = torch.nn.BatchNorm2d(8)l = [conv,bn]for module in l: print('{}.weight.data.size()'.format(str(module.__class__.__name__)),module.weight.data.size())原创 2020-06-13 13:07:54 · 682 阅读 · 0 评论 -
pytorch计算module的参数量
import torchconv = torch.nn.Conv2d(1,8,(2,3))bn = torch.nn.bBatchNorm2d(8)l = [conv,bn]for module in l: params = 0 for param_ in module.parameters(): print(param_) params += param_.view(-1).size(0) print(module,params)可以看到原创 2020-06-13 12:09:27 · 591 阅读 · 0 评论