def METHOD_NAME(self, batch, stage):
    "Given an input batch it computes the enhanced signal"
    batch = batch.to(self.device)
    wavs, lens = batch.sig
    noisy = self.hparams.add_noise(wavs, lens).unsqueeze(-1)
    enhanced = self.modules.generator(noisy)
    return enhanced