Skip to content

GitLab

Commit d5637696 authored by kwu's avatar kwu Committed by Francisco Massa
Browse files

Changing to AdaptiveAvgPool2d on DenseNet (#672) 

* Changing to AdaptiveAvgPool2d

* add whitespace
parent 50d9dc5f
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
torchvision/models/densenet.py
View file @ d5637696
...@@ -112,7 +112,7 @@ class DenseNet(nn.Module): ...@@ -112,7 +112,7 @@ class DenseNet(nn.Module):
def forward(self, x): def forward(self, x):
features = self.features(x) features = self.features(x)
out = F.relu(features, inplace=True) out = F.relu(features, inplace=True)
out = F.avg_pool2d(out, kernel_size=7, stride=1).view(features.size(0), -1) out = F.adaptive_avg_pool2d(out, (1, 1)).view(features.size(0), -1)
out = self.classifier(out) out = self.classifier(out)
return out return out
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment