方法很简单，你只需要将模型最后的全连接层改成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。
Pytorch如何用预训练模型提取图像特征
免责声明！
本站转载的文章为个人学习借鉴使用，本站对版权不负任何法律责任。如果侵犯了您的隐私权益，请联系本站邮箱yoyou2525@163.com删除。