hub / github.com/NVlabs/DiffPure / get_pc_sampler

Function get_pc_sampler

score_sde/sampling.py:355–411 · view source on GitHub ↗

Create a Predictor-Corrector (PC) sampler. Args: sde: An `sde_lib.SDE` object representing the forward SDE. shape: A sequence of integers. The expected shape of a single sample. predictor: A subclass of `sampling.Predictor` representing the predictor algorithm. corrector: A subcla

(sde, shape, predictor, corrector, inverse_scaler, snr,
                   n_steps=1, probability_flow=False, continuous=False,
                   denoise=True, eps=1e-3, device='cuda')

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

get_sampling_fnFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected