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

Method sample

guided_diffusion/resample.py:42–58  ·  view source on GitHub ↗

Importance-sample timesteps for a batch. :param batch_size: the number of timesteps. :param device: the torch device to save to. :return: a tuple (timesteps, weights): - timesteps: a tensor of timestep indices. - weights: a tensor o

(self, batch_size, device)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers

nothing calls this directly

Calls 1

weightsMethod · 0.95

Tested by

no test coverage detected