amend

pytorch · Aug 5, 2021 · e3a84ef · e3a84ef
1 parent 1e72acb
commit e3a84ef
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchvision/models/inception.py b/torchvision/models/inception.py
@@ -120,7 +120,7 @@ def __init__(
         if init_weights:
             for m in self.modules():
                 if isinstance(m, nn.Conv2d) or isinstance(m, nn.Linear):
-                    stddev = float(m.stddev) if hasattr(m, 'stddev') else 0.1
+                    stddev = m.stddev if hasattr(m, 'stddev') else 0.1  # type: ignore[assignment]
                     torch.nn.init.trunc_normal_(m.weight, mean=0.0, std=stddev, a=-2, b=2)
                 elif isinstance(m, nn.BatchNorm2d):
                     nn.init.constant_(m.weight, 1)