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

Method update_with_local_losses

guided_diffusion/resample.py:79–112  ·  view source on GitHub ↗

Update the reweighting using losses from a model. Call this method from each rank with a batch of timesteps and the corresponding losses for each of those timesteps. This method will perform synchronization to make sure all of the ranks maintain the exact sa

(self, local_ts, local_losses)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

forward_backwardMethod · 0.80

Calls 1

Tested by

no test coverage detected