self.vgg = models.vgg16()
# Remove fc8
self.vgg.classifier = nn.Sequential(*list(self.vgg.classifier._modules.values())[:-1])
list变量前面加星号,字典变量前面加两个星号
列表前面加星号作用是将列表解开成两个独立的参数,传入函数,
字典前面加两个星号,是将字典解开成独立的元素作为形参。
self.vgg = models.vgg16()
# Remove fc8
self.vgg.classifier = nn.Sequential(*list(self.vgg.classifier._modules.values())[:-1])
列表前面加星号作用是将列表解开成两个独立的参数,传入函数,
字典前面加两个星号,是将字典解开成独立的元素作为形参。