def __init__()

in sing/ae/models.py [0:0]

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

    def __init__(self,
                 channels=4096,
                 stride=256,
                 dimension=128,
                 kernel_size=1024,
                 rewrite_layers=2,
                 window_name="hann",
                 squared_window=True):
        super(ConvolutionalEncoder, self).__init__()
        layers = []
        conv = nn.Conv1d(
            in_channels=1,
            out_channels=channels,
            kernel_size=kernel_size,
            stride=stride)
        if window_name is not None:
            conv = WindowedConv1d(conv, window_name, squared_window)
        layers.extend([conv, nn.ReLU()])
        for rewrite in range(rewrite_layers):
            layers.extend([
                nn.Conv1d(
                    in_channels=channels, out_channels=channels,
                    kernel_size=1),
                nn.ReLU()
            ])

        layers.append(
            nn.Conv1d(
                in_channels=channels, out_channels=dimension, kernel_size=1))
        self.layers = nn.Sequential(*layers)