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())
查看Conv2d,BN两类module的参数大小
最新推荐文章于 2024-03-20 10:52:48 发布