pytorch中存储各层权重参数时的命名规则,为什么有些层的名字中带module.
1,对于__init__中使用self定义的变量会使用这个变量名作为存储时的名字 如下: (1)self.conv1 = nn.Conv2d(3, 12, kernel_size=3, stride=1, padding=1) 卷积层有两个参数:权重和偏移项,对应的名称为conv1.weight、conv1.bias (2)self.bn1 = nn.BatchNorm2d(12) BN层有5个参数:bn1.weight、bn...
原创
2020-05-20 11:26:55 ·
8962 阅读 ·
3 评论