def master_params_to_model_params()

in improved_diffusion/fp16_util.py [0:0]

• 6 lines of code
• 3 McCabe index (conditional complexity)

def master_params_to_model_params(model_params, master_params):
    """
    Copy the master parameter data back into the model parameters.
    """
    # Without copying to a list, if a generator is passed, this will
    # silently not copy any parameters.
    model_params = list(model_params)

    for param, master_param in zip(
        model_params, unflatten_master_params(model_params, master_params)
    ):
        param.detach().copy_(master_param)