vgg去掉最后一层fc8

self.vgg = models.vgg16()

    # Remove fc8

self.vgg.classifier = nn.Sequential(*list(self.vgg.classifier._modules.values())[:-1])





list变量前面加星号,字典变量前面加两个星号

列表前面加星号作用是将列表解开成两个独立的参数,传入函数,

字典前面加两个星号,是将字典解开成独立的元素作为形参。


最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。