Pytorch中VGG如何实现修改最后一层FC

这篇文章主要介绍Pytorch中VGG如何实现修改最后一层FC，文中介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们一定要看完！

That's because vgg19 doesn't have a fc member variable. Instead, it has a

 (classifier): Sequential (
 (0): Dropout (p = 0.5)
 (1): Linear (25088 -> 4096)
 (2): ReLU (inplace)
 (3): Dropout (p = 0.5)
 (4): Linear (4096 -> 4096)
 (5): ReLU (inplace)
 (6): Linear (4096 -> 100)
 )

To replace the last linear layer, a temporary solution would be

vgg19.classifier._modules['6'] = nn.Linear(4096, 8)

以上是“Pytorch中VGG如何实现修改最后一层FC”这篇文章的所有内容，感谢各位的阅读！希望分享的内容对大家有帮助，更多相关知识，欢迎关注行业资讯频道！