pytorch 更改预训练模型网络结构的方法

一个继承nn.module的model它包含一个叫做children()的函数，这个函数可以用来提取出model每一层的网络结构，在此基础上进行修改即可，修改方法如下(去除后两层)： resnet_layer = nn.Sequential(*list(model.children())[:-2]) 那么，接下来就可以构建我们的网络了： class Net(nn.Module): def __init__(self , model): super(Net, self).__init__() #取掉model的后两层 self.resnet_layer = nn.