浩晨众云网站建设,新征程启航
为企业提供网站建设、域名注册、服务器等服务
这篇文章主要介绍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”这篇文章的所有内容,感谢各位的阅读!希望分享的内容对大家有帮助,更多相关知识,欢迎关注创新互联行业资讯频道!