方法很简单，你只需要将模型最后的全连接层改成Dropout即可。
import torch from torchvision import models # load data x, y = get_data() ... model = models.VGG16(pretrained=True) model.classifier = torch.nn.Dropout() feature = model(x)
over。