def update_init_info(module, init_info):
    'Update the `_params_init_info` in the module if the value of parameters\n    are changed.\n\n    Args:\n        module (obj:`nn.Module`): The module of PyTorch with a user-defined\n            attribute `_params_init_info` which records the initialization\n            information.\n        init_info (str): The string that describes the initialization.\n    '
    assert hasattr(module, '_params_init_info'), f'Can not find `_params_init_info` in {module}'
    for (name, param) in module.named_parameters():
        assert (param in module._params_init_info), f'Find a new :obj:`Parameter` named `{name}` during executing the `init_weights` of `{module.__class__.__name__}`. Please do not add or replace parameters during executing the `init_weights`. '
        mean_value = param.data.mean()
        if (module._params_init_info[param]['tmp_mean_value'] != mean_value):
            module._params_init_info[param]['init_info'] = init_info
            module._params_init_info[param]['tmp_mean_value'] = mean_value
Update the `_params_init_info` in the module if the value of parameters
are changed.

Args:
    module (obj:`nn.Module`): The module of PyTorch with a user-defined
        attribute `_params_init_info` which records the initialization
        information.
    init_info (str): The string that describes the initialization.
def update_init_info(module, init_info):
    'Update the `_params_init_info` in the module if the value of parameters\n    are changed.\n\n    Args:\n        module (obj:`nn.Module`): The module of PyTorch with a user-defined\n            attribute `_params_init_info` which records the initialization\n            information.\n        init_info (str): The string that describes the initialization.\n    '
    assert hasattr(module, '_params_init_info'), f'Can not find `_params_init_info` in {module}'
    for (name, param) in module.named_parameters():
        assert (param in module._params_init_info), f'Find a new :obj:`Parameter` named `{name}` during executing the `init_weights` of `{module.__class__.__name__}`. Please do not add or replace parameters during executing the `init_weights`. '
        mean_value = param.data.mean()
        if (module._params_init_info[param]['tmp_mean_value'] != mean_value):
            module._params_init_info[param]['init_info'] = init_info
            module._params_init_info[param]['tmp_mean_value'] = mean_value