MCPcopy Create free account
hub / github.com/ImprintLab/MedSegDiff / model_fn

Function model_fn

guided_diffusion/dpm_solver.py:278–299  ·  view source on GitHub ↗

The noise predicition model function that is used for DPM-Solver.

(x, t_continuous)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

__init__Method · 0.85

Calls 4

noise_pred_fnFunction · 0.85
get_model_input_timeFunction · 0.85
cond_grad_fnFunction · 0.85
marginal_stdMethod · 0.80

Tested by

no test coverage detected