7c3c41e93c1ad3dacbc2cbbaab9942a8d96c37bc,jiant/utils/torch_utils.py,,copy_state_dict,#Any#Any#,41

Before Change

    if target_device is None:
        return copied_state_dict
    else:
        return {k: v.to(target_device) for k, v in copied_state_dict.items()}


def get_parent_child_module_list(model):
    ls = []

After Change

        new_state_dict = {}
        unique_dict = {}
        for k, v in copied_state_dict.items():
            unique_key = tuple(v.shape), v.data_ptr()
            if unique_key not in unique_dict:
                unique_dict[unique_key] = v.to(target_device)
            // Create a view
            new_state_dict[k] = unique_dict[unique_key][:]

Instances