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()