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

Function model_grads_to_master_grads

guided_diffusion/fp16_util.py:52–62  ·  view source on GitHub ↗

Copy the gradients from the model parameters into the master parameters from make_master_params().

(param_groups_and_shapes, master_params)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

_optimize_fp16Method · 0.85

Calls 1

param_grad_or_zerosFunction · 0.85

Tested by

no test coverage detected