查看Conv2d,BN两类module的参数大小

import torch
conv = 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())