diffusion_models.models.fp16_util.model_grads_to_master_grads

diffusion_models.models.fp16_util.model_grads_to_master_grads(model_params, master_params)[source]

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