Fix location of dropout in VGG (#107)

torchvision/models/__init__.py

@@ -37,10 +37,10 @@ ResNet-101               22.63           6.44
 ResNet-152               21.69           5.94
 Inception v3             22.55           6.44
 AlexNet                  43.45           20.91
-VGG-11                   32.14           12.12
-VGG-13                   31.04           11.40
-VGG-16                   29.11           10.17
-VGG-19                   28.42           9.69
+VGG-11                   30.98           11.37
+VGG-13                   30.07           10.75
+VGG-16                   28.41           9.62
+VGG-19                   27.62           9.12
 SqueezeNet 1.0           41.90           19.58
 SqueezeNet 1.1           41.81           19.38
 ======================== =============   =============

torchvision/models/vgg.py

@@ -10,10 +10,10 @@ __all__ = [
 
 
 model_urls = {
-    'vgg11': 'https://download.pytorch.org/models/vgg11-fb7e83b2.pth',
-    'vgg13': 'https://download.pytorch.org/models/vgg13-58758d87.pth',
-    'vgg16': 'https://download.pytorch.org/models/vgg16-82412952.pth',
-    'vgg19': 'https://download.pytorch.org/models/vgg19-341d7465.pth',
+    'vgg11': 'https://download.pytorch.org/models/vgg11-bbd30ac9.pth',
+    'vgg13': 'https://download.pytorch.org/models/vgg13-c768596a.pth',
+    'vgg16': 'https://download.pytorch.org/models/vgg16-397923af.pth',
+    'vgg19': 'https://download.pytorch.org/models/vgg19-dcbb9e9d.pth',
 }
 
 
@@ -23,12 +23,12 @@ class VGG(nn.Module):
         super(VGG, self).__init__()
         self.features = features
         self.classifier = nn.Sequential(
-            nn.Dropout(),
             nn.Linear(512 * 7 * 7, 4096),
             nn.ReLU(True),
             nn.Dropout(),
             nn.Linear(4096, 4096),
             nn.ReLU(True),
+            nn.Dropout(),
             nn.Linear(4096, 1000),
         )
         self._initialize_weights()