Function vgather

PATH/core/distributed_utils.py:1129–1130

(x_list, x)

    x.update(y.item())

def vgather(x_list, x):
    allgather(x_list, torch.Tensor([x]).cuda())

def reduce_dict(input_dict, task_size, task_rank, group=None, average=True):
    """