add comment (#1932)

pytorch · Mar 4, 2020 · c3e2b01 · c3e2b01
1 parent 3b0b201
commit c3e2b01
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/torchvision/models/googlenet.py b/torchvision/models/googlenet.py
@@ -224,6 +224,8 @@ def __init__(self, in_channels, ch1x1, ch3x3red, ch3x3, ch5x5red, ch5x5, pool_pr
 
         self.branch3 = nn.Sequential(
             conv_block(in_channels, ch5x5red, kernel_size=1),
+            # Here, kernel_size=3 instead of kernel_size=5 is a known bug.
+            # Please see https://github.com/pytorch/vision/issues/906 for details.
             conv_block(ch5x5red, ch5x5, kernel_size=3, padding=1)
         )