def __init__()

in codes/models.py [0:0]


    def __init__(self):
        super(ANN, self).__init__()
        self.l1 = nn.Linear(1 * 150 * 2048 , 32)
        self.l2 = nn.Linear(32 , 1)
        self.act= nn.ReLU()