pytorch怎么使用定义好的模型的一部分

Encoder代码为：

class Encoder(nn.Module):  #输入图片的大小isize、噪声的维度nz=100、输入图片的通道nc=3、ndf=64、
    def __init__(self,isize,nz,nc,ndf,ngpu,n_exter_layers=0,add_final_conv=True):
        super(Encoder,self).__init__()
        self.ngpu=ngpu
        # 必须为16倍数
        assert isize % 16==0,"isize has to be a multiple of 16"

        main=nn.Sequential()
        # 图片的高宽缩小一倍
        main.add_module('initial-conv-{0}-{1}'.format(nc,ndf),nn.Conv2d(nc,ndf,4,2,1,bias=False))
        main.add_module('initial-relu-{0}'.format(ndf),nn.LeakyReLU(0.2,inplace=True))
        csize,cndf=isize/2,ndf

        for t in range(n_exter_layers): #在这里面特征宽高不变，通道数也不变
            main.add_module('extra-layers-{0}-{1}-conv'.format(t,cndf),nn.Conv2d(cndf,cndf,3,1,1,bias=False))
            main.add_module('extra-layers-{0}-{1}-batchnorm'.format(t,cndf),nn.BatchNorm2d(cndf))
            main.add_module('extra-layers-{0}-{1}-relu'.format(t,cndf),nn.LeakyReLU(0.2,inplace=True))

        # 在特征高宽仍大于4时，就添加缩小一倍高宽，通道增加一倍的卷积块
        while csize>4:
            in_feat = cndf

            out_feat = cndf * 2

            main.add_module('pyramid-{0}-{1}-conv'.format(in_feat, out_feat),nn.Conv2d(in_feat, out_feat, 4, 2, 1, bias=False))

            main.add_module('pyramid-{0}-batchnorm'.format(out_feat),nn.BatchNorm2d(out_feat))

            main.add_module('pyramid-{0}-relu'.format(out_feat),nn.LeakyReLU(0.2, inplace=True))

            cndf = cndf * 2

            csize = csize / 2

        # 最后一层卷积，将4*4变为1*1,得到nz = 100的噪声
        if add_final_conv:

            main.add_module('final-{0}-{1}-conv'.format(cndf, 1),nn.Conv2d(cndf, nz, 4, 1, 0, bias=False))
            self.main=main
    
    def forward(self,input):
        if self.ngpu>1:
             output=nn.parallel.data_parallel(self.main,input,range(self.ngpu)) #在多个gpu上运行模型，并行计算
        else:
            output=self.main(input)
        
        return output  #如果输入的大小是3×32×32，最后的输出是100×1×1.

判别器为：

 #定义判别器D （编码器）     
class NetD(nn.Module):
    def __init__(self, opt):
        super(NetD, self).__init__()
        
        # 第二个参数是1 因为判别器最后输出一个数
        # 不过编码器在生成器里的时候 
        # 这个参数是100 因为它要把图片下采样成100×1×1的向量
        model = Encoder(opt.isize, 1, opt.nc, opt.ndf, opt.ngpu, opt.extralayers)  

        layers = list(model.main.children())

        '''layers的输出如下：
        [
        Conv2d(3, 64, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1), bias=False),

        LeakyReLU(negative_slope=0.2, inplace),

        Conv2d(64, 128, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1), bias=False),

        BatchNorm2d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True),

        LeakyReLU(negative_slope=0.2, inplace),

        Conv2d(128, 256, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1), bias=False),

        BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True),

        LeakyReLU(negative_slope=0.2, inplace),

        Conv2d(256, 1, kernel_size=(4, 4), stride=(1, 1), bias=False)] 因为132行定义的nz参数是1，所以经过这层之后输出的大小是1×1×1
        '''

        self.features = nn.Sequential(*layers[:-1])
        # self.features的内容为除了最后一层的前8层
        # nn.Sequential函数里面的参数一定是Module的子类，而list不是一个模块子类，所以不能当做参数
        # 当然model.children()也一样不是一个模块子类，只有他们里面的值才是
        # 这里的*就起了作用，将list或者children的内容迭代地一个一个的传进去。

        #生成一个分类器模块
        self.classifier = nn.Sequential(layers[-1])
        #self.classifier的内容为Conv2d(256, 1, kernel_size=(4, 4), stride=(1, 1), bias=False)最后一层
        #并在后面添加子模块sigmoid
        self.classifier.add_module('Sigmoid', nn.Sigmoid())

    def forward(self, x):

        features = self.features(x) #图片通过前8层之后的结果256×4×4，前提是输入的图片的大小是32

        features = features

        classifier = self.classifier(features)#此时的结果是1×1×1，值在[0，1]，因为经过了sigmoid

        classifier = classifier.view(-1, 1).squeeze(1)
        #a=torch.ones([1,1,1])  即a=tensor([[[ 1.]]])   再a.view(-1,1) 变成tensor([[ 1.]])  再加一个squeeze就是
        # a.view(-1,1).squeeze(1) 结果是tensor([ 1.])，squeeze里的值是0是1随意，只要去掉一个维度上的就可以

        # 因此返回得到一个判别值classifier 和一个大小为256×4×4的特征值
        return classifier, features

重点在：

• layers = list(model.main.children())
• self.features = nn.Sequential(*layers[:-1]) ：使用除了最后一层的前面所有层
• self.classifier = nn.Sequential(layers[-1])：仅使用最后一层

这里可见同样的模型结构我们在Encoder.py中已经定义过一遍了，在判别其中实在不想再定义一遍，那我们就能够使用model.main.children()来获得模块中的子模块，然后将其转成列表形式，然后就能够根据想要的部分来进行处理了