langevin.py

Source: LION/reconstructors/diffusion/langevin.py

Annealed Langevin sampling for diffusion reconstructors.

class LION.reconstructors.diffusion.langevin.AnnealedLangevin[source]

Bases: object

Reconstruct with the annealed Langevin sampler, optionally using DDNM.

langevin(measurement, params, *, prog_bar, generator)[source]

Reconstruct a measurement using annealed Langevin dynamics.

Parameters:

  • measurement (Tensor)

  • prog_bar (bool)

  • generator (Generator | None)

Return type:

Tensor