hub / github.com/NVlabs/DiffPure / reverse

Method reverse

score_sde/sde_lib.py:79–117 · view source on GitHub ↗

Create the reverse-time SDE/ODE. Args: score_fn: A time-dependent score-based model that takes x and t and returns the score. probability_flow: If `True`, create the reverse-time ODE used for probability flow sampling.

(self, score_fn, probability_flow=False)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 2

__init__Method · 0.80

drift_fnFunction · 0.80

Calls 1

RSDEClass · 0.85

Tested by

no test coverage detected