Fixes #1797 by adding an init_weights keyword argument to Inception3 (#1832)

torchvision/models/inception.py

@@ -65,7 +65,7 @@ def inception_v3(pretrained=False, progress=True, **kwargs):
 class Inception3(nn.Module):
 
     def __init__(self, num_classes=1000, aux_logits=True, transform_input=False,
-                 inception_blocks=None):
+                 inception_blocks=None, init_weights=True):
         super(Inception3, self).__init__()
         if inception_blocks is None:
             inception_blocks = [
@@ -102,7 +102,7 @@ class Inception3(nn.Module):
         self.Mixed_7b = inception_e(1280)
         self.Mixed_7c = inception_e(2048)
         self.fc = nn.Linear(2048, num_classes)
-
+        if init_weights:
             for m in self.modules():
                 if isinstance(m, nn.Conv2d) or isinstance(m, nn.Linear):
                     import scipy.stats as stats