def __init__()

in models.py [0:0]


    def __init__(self, num_channels, kernel_size, stride, pool_size, bias=True, normalize=False):
        super(Extractor, self).__init__()
        self.normalize = normalize
        self.pool_size = pool_size
        conv_layers = []
        assert(len(num_channels) >= 2)
        self.conv_layers = nn.ModuleList([nn.Conv2d(num_channels[i], num_channels[i+1],
            kernel_size, stride, bias=bias) for i in range(len(num_channels)-1)])