59d4bbaaf06850e2dfb357855f50b05f6c9aadb1,syft/frameworks/torch/fl/utils.py,,scale_model,#Any#Any#,53

Before Change

    
    params = model.named_parameters()
    dict_params = dict(params)
    with torch.no_grad():
        for name, param in dict_params.items():
            dict_params[name].set_(dict_params[name].data * scale)
    return model

After Change

    with torch.no_grad():
        for name in params:
            params[name] = params[name].type_as(scale) * scale
    scaled_model = copy.deepcopy(model)
    scaled_model.load_state_dict(params, strict=False)
    return scaled_model

Instances