def __init__()

in archs/models.py [0:0]

• 19 lines of code
• 5 McCabe index (conditional complexity)

    def __init__(self, num_layers, num_modules_per_layer, in_dim, inter_dim):
        """TODO: to be defined1. """
        nn.Module.__init__(self)
        self.start_modules = [nn.Sequential()]
        self.layer1 = [[
            nn.Sequential(
                nn.Linear(in_dim, inter_dim), nn.BatchNorm1d(inter_dim),
                nn.ReLU()) for _ in range(num_modules_per_layer)
        ]]
        if num_layers > 1:
            self.layer2 = [[
                nn.Sequential(
                    nn.BatchNorm1d(inter_dim), nn.Linear(inter_dim, inter_dim),
                    nn.BatchNorm1d(inter_dim), nn.ReLU())
                for _m in range(num_modules_per_layer)
            ] for _l in range(num_layers - 1)]
        self.avgpool = nn.Sequential()
        self.fc = [
            nn.Sequential(nn.BatchNorm1d(inter_dim), nn.Linear(inter_dim, 1))
        ]