def __init__()

in src/models.py [0:0]

• 9 lines of code
• 3 McCabe index (conditional complexity)

    def __init__(self, z_dim, channels=64, blocks=3, layers_per_block=10, conv_len=2):
        super().__init__()
        self.layers = []
        self.rectv_field = 1
        for b in range(blocks):
            for l in range(layers_per_block):
                self.layers += [HyperConvBlock(channels, channels, z_dim, kernel_size=conv_len, dilation=2**l)]
                self.rectv_field += self.layers[-1].receptive_field() - 1
        self.layers = nn.ModuleList(self.layers)