espnet2.enh.diffusion.sampling.correctors.NoneCorrector

source

class espnet2.enh.diffusion.sampling.correctors.NoneCorrector(*args, **kwargs)

Bases: Corrector

An empty corrector that does nothing.

update_fn(x, t, *args)

One update of the corrector.

• Parameters:
  • x – A PyTorch tensor representing the current state
  • t – A PyTorch tensor representing the current time step.
  • *args – Possibly additional arguments, in particular y for OU processes
• Returns: A PyTorch tensor of the next state. x_mean: A PyTorch tensor. The next state without random noise.

  Useful for denoising.

• Return type: x