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 d941e7e commit e010fe8Copy full SHA for e010fe8
models/mobilenetv2.py
@@ -72,8 +72,8 @@ def forward(self, x):
72
out = F.relu(self.bn1(self.conv1(x)))
73
out = self.layers(out)
74
out = F.relu(self.bn2(self.conv2(out)))
75
- # NOTE: change pooling kernel_size 7 -> 2 for CIFAR10
76
- out = F.avg_pool2d(out, 2)
+ # NOTE: change pooling kernel_size 7 -> 4 for CIFAR10
+ out = F.avg_pool2d(out, 4)
77
out = out.view(out.size(0), -1)
78
out = self.linear(out)
79
return out
0 commit comments