Pytorch固定某些層的操作

1. model = models.resnet18(pretrained=False,num_classes=CIFAR10_num_classes) def my_forward(model, x): mo = nn.Sequential(*list(model.children())[:-1]) feature = mo(x) feature = feature.view(x.size(0), -1) output= model.fc(feature) return feature, output


2. 在模型的forward函數的全連接層前面加上self.feature = x, 即可.