MCPcopy Create free account
hub / github.com/NVlabs/DiffPure / model_fn

Function model_fn

score_sde/models/utils.py:107–123  ·  view source on GitHub ↗

Compute the output of the score-based model. Args: x: A mini-batch of input data. labels: A mini-batch of conditioning variables for time steps. Should be interpreted differently for different models. Returns: A tuple of (model output, new

(x, labels)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 2

loss_fnFunction · 0.85
score_fnFunction · 0.85

Calls 2

trainMethod · 0.80
evalMethod · 0.45

Tested by

no test coverage detected