def METHOD_NAME(self, batch, stage):
    "Given an input batch it computes the phoneme probabilities."
    batch = batch.to(self.device)
    wavs, wav_lens = batch.sig
    wavs = self.modules.augmentation(wavs, wav_lens)
    feats = self.hparams.compute_features(wavs)
    feats = self.modules.normalize(feats, wav_lens)
    out = self.modules.model(feats)
    out = self.modules.output(out)
    pout = self.hparams.log_softmax(out)
    return pout, wav_lens