dps_langevin.py

Source: LION/reconstructors/diffusion/dps_langevin.py

Diffusion posterior sampling with Langevin dynamics.

class LION.reconstructors.diffusion.dps_langevin.DPSLangevin[source]

Bases: object

Reconstruct with diffusion posterior sampling and Langevin updates.

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

Reconstruct a measurement using the DPS-Langevin sampler.

Parameters:

  • measurement (Tensor)

  • prog_bar (bool)

  • generator (Generator | None)

Return type:

Tensor