hub / github.com/ImprintLab/MedSegDiff / get_time_steps

Method get_time_steps

guided_diffusion/dpm_solver.py:423–449 · view source on GitHub ↗

Compute the intermediate time steps for sampling. Args: skip_type: A `str`. The type for the spacing of the time steps. We support three types: - 'logSNR': uniform logSNR for the time steps. - 'time_uniform': uniform time for the time steps. (**Rec

(self, skip_type, t_T, t_0, N, device)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 2

get_orders_and_timesteps_for_singlestep_solverMethod · 0.95

sampleMethod · 0.95

Calls 2

marginal_lambdaMethod · 0.80

inverse_lambdaMethod · 0.80

Tested by

no test coverage detected