From 5f6d19a67bb3503816b8b822f44fd31ae189e492 Mon Sep 17 00:00:00 2001 From: quinwu Date: Tue, 27 Nov 2018 15:10:57 +0800 Subject: [PATCH 1/2] Changing to AdaptiveAvgPool2d --- torchvision/models/densenet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchvision/models/densenet.py b/torchvision/models/densenet.py index 0a9376d1b7a..94f81f0d300 100644 --- a/torchvision/models/densenet.py +++ b/torchvision/models/densenet.py @@ -112,7 +112,7 @@ def __init__(self, growth_rate=32, block_config=(6, 12, 24, 16), def forward(self, x): features = self.features(x) 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) return out From 952ca34e10f500417b3a02b68d2871332f058a46 Mon Sep 17 00:00:00 2001 From: quinwu Date: Tue, 27 Nov 2018 15:42:46 +0800 Subject: [PATCH 2/2] add whitespace --- torchvision/models/densenet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchvision/models/densenet.py b/torchvision/models/densenet.py index 94f81f0d300..43f8e7c3eba 100644 --- a/torchvision/models/densenet.py +++ b/torchvision/models/densenet.py @@ -112,7 +112,7 @@ def __init__(self, growth_rate=32, block_config=(6, 12, 24, 16), def forward(self, x): features = self.features(x) out = F.relu(features, inplace=True) - out = F.adaptive_avg_pool2d(out, (1,1)).view(features.size(0), -1) + out = F.adaptive_avg_pool2d(out, (1, 1)).view(features.size(0), -1) out = self.classifier(out) return out