hub / github.com/NVlabs/DiffPure / sample

Method sample

guided_diffusion/resample.py:50–66 · 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 3

attackMethod · 0.80

display_adversarial_2rowFunction · 0.80

forward_backwardMethod · 0.80

Calls 1

weightsMethod · 0.95

Tested by

no test coverage detected