We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 0c5598c commit 2acfb23Copy full SHA for 2acfb23
torch/nn/modules/instancenorm.py
@@ -26,7 +26,7 @@ def forward(self, input):
26
27
out = F.batch_norm(
28
input_reshaped, running_mean, running_var, weight, bias,
29
- self.training, self.momentum, self.eps)
+ True, self.momentum, self.eps)
30
31
# Reshape back
32
self.running_mean.copy_(running_mean.view(b, c).mean(0))
0 commit comments