Skip to content

GitLab

Unverified Commit c3e2b018 authored by Philip Meier's avatar Philip Meier Committed by GitHub
Browse files

add comment (#1932)

parent 3b0b201a
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
torchvision/models/googlenet.py
View file @ c3e2b018
...@@ -224,6 +224,8 @@ class Inception(nn.Module): ...@@ -224,6 +224,8 @@ class Inception(nn.Module):
self.branch3 = nn.Sequential( self.branch3 = nn.Sequential(
conv_block(in_channels, ch5x5red, kernel_size=1), 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) conv_block(ch5x5red, ch5x5, kernel_size=3, padding=1)
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment