Skip to content

Commit e010fe8

Browse files
author
kuangliu
committed
Fix typo (pytorch#29)
1 parent d941e7e commit e010fe8

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

models/mobilenetv2.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -72,8 +72,8 @@ def forward(self, x):
7272
out = F.relu(self.bn1(self.conv1(x)))
7373
out = self.layers(out)
7474
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)
75+
# NOTE: change pooling kernel_size 7 -> 4 for CIFAR10
76+
out = F.avg_pool2d(out, 4)
7777
out = out.view(out.size(0), -1)
7878
out = self.linear(out)
7979
return out

0 commit comments

Comments
 (0)