DiffusionIterator

class deepinv.sampling.DiffusionIterator(cur_params=None, clip=None)

Bases: SamplingIterator

Helper class used by deepinv.sampling.DiffusionSampler to interface diffusion models with the deepinv.sampling.BaseSampling framework.

Note

Users should typically interact with deepinv.sampling.DiffusionSampler rather than this class directly.