Pytorch加载预训练模型前n层
Posted dyclown
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了Pytorch加载预训练模型前n层相关的知识，希望对你有一定的参考价值。
import torch.nn as nn
import torchvision.models as models
class resnet(nn.Module):
def __init__(self):
super(resnet,self).__init__()
self.model = models.resnet18(pretrained=True)
self.encoder = nn.Sequential(*list(self.model.children())[:-1])
def forward(self,x):
y = self.encoder(x)
return y
net = resnet()以上是关于Pytorch加载预训练模型前n层的主要内容，如果未能解决你的问题，请参考以下文章