def forward(self, batch):
    (x, _, _) = batch
    x = x.float().transpose(1, 2)
    x = self.spec(x)
    x = self.net['convs'](x)
    x = x.view(x.size(0), (- 1))
    x = self.net['dense'](x)
    x = F.log_softmax(x, dim=1)
    return x
def forward(self, batch):
    (x, _, _) = batch
    x = x.float().transpose(1, 2)
    x = self.spec(x)
    x = self.net['convs'](x)
    x = x.view(x.size(0), (- 1))
    x = self.net['dense'](x)
    x = F.log_softmax(x, dim=1)
    return x