Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

FC layers for VGG? #41

Open
gabewb opened this issue Mar 4, 2021 · 0 comments
Open

FC layers for VGG? #41

gabewb opened this issue Mar 4, 2021 · 0 comments

Comments

@gabewb
Copy link

gabewb commented Mar 4, 2021

Reading vgg.py, I'm confused about the last layers of the network.

My understanding is that VGG normally ends with a few fully connected layers and a softmax, but here we end with:

layers += [nn.AvgPool2d(kernel_size=1, stride=1)]
and then
self.classifier = nn.Linear(512, 10)

My questions are:

  1. What's the deal with AvgPool2d with kernel size 1? Seems like it should be a no-op?
  2. Why were the fully connected layers removed? Is this an adjustment for the easier task of CIFAR-10, as compared to ImageNet?
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@gabewb