方法很简单，你只需要将模型最后的全连接层改成Sequential即可。
import torch
from torchvision import models
# load data
x, y = get_data()
...
model = models.VGG16(pretrained=True)
model.classifier = torch.nn.Sequential()
model.eval()
feature = model(x)
over。
相关文章：
猜你喜欢
相关资源
相似解决方案
热门标签