in src/image_gen_aux/preprocessors/teed/teed.py [0:0]
def __init__(self, num_layers, input_features, out_features):
super(_DenseBlock, self).__init__()
for i in range(num_layers):
layer = _DenseLayer(input_features, out_features)
self.add_module("denselayer%d" % (i + 1), layer)
input_features = out_features