def forward()

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