如何从vgg19中删除自适应平均池层？

Question

我已经加载了vgg19的预训练模型。如何删除分类器之前存在的自适应平均池层？

Answer 1

如果使用的是PyTorch和torchVision模型，则可以像这样禁用最后一个maxpool层的使用-

nn.Sequential(*list(vgg.features._modules.values())[:-1])

这里所有层都在一个数组中，并且不包括最后一个元素就可以解决问题。