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)])