hub / github.com/ReinFlow/ReinFlow / reverse

Method reverse

model/diffusion/sde_lib.py:135–177 · 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 1

drift_fnFunction · 0.80

Calls 1

RSDEClass · 0.85

Tested by

no test coverage detected