小编yjy*_*131的帖子

BatchNorm2d 的 running_mean / running_var 在 PyTorch 中意味着什么?

我想知道我可以从running_mean和处running_var拨打电话nn.BatchNorm2d

示例代码在这里,其中 bn 表示nn.BatchNorm2d

vector = torch.cat([
    torch.mean(self.conv3.bn.running_mean).view(1), torch.std(self.conv3.bn.running_mean).view(1),
    torch.mean(self.conv3.bn.running_var).view(1), torch.std(self.conv3.bn.running_var).view(1),
    torch.mean(self.conv5.bn.running_mean).view(1), torch.std(self.conv5.bn.running_mean).view(1),
    torch.mean(self.conv5.bn.running_var).view(1), torch.std(self.conv5.bn.running_var).view(1)
])
Run Code Online (Sandbox Code Playgroud)

我无法弄清楚Pytorch 官方文档和用户社区中的running_mean和是什么意思。running_var

nn.BatchNorm2.running_mean和是什么nn.BatchNorm2.running_var意思?

mean variance deep-learning pytorch batch-normalization

3
推荐指数
1
解决办法
4677
查看次数