def METHOD_NAME(self, batch, stage):
    "Given an input batch it computes the output probabilities."
    batch = batch.to(self.device)
    wavs, wav_lens = batch.sig
    phns_bos, _ = batch.phn_encoded_bos
    feats = self.hparams.compute_features(wavs)
    feats = self.modules.mean_var_norm(feats, wav_lens)
    x = self.modules.enc(feats)
    # Prepend bos token at the beginning
    e_in = self.modules.emb(phns_bos)
    h, w = self.modules.dec(e_in, x, wav_lens)
    logits = self.modules.lin(h)
    outputs = self.hparams.softmax(logits)
    if stage != sb.Stage.TRAIN:
        seq, _ = self.hparams.searcher(x, wav_lens)
        return outputs, seq
    return outputs