in codes/models.py [0:0]
def forward(self, x1): x1 = x1.view(-1, 1 * 150 * 2048) x1 = self.act(self.l1(x1)) x2 = self.l2(x1) return x2