Function to_python_float

experiments/ddp.py:89–93

(t)

    random.seed(args.proc_id)

def to_python_float(t):
    if hasattr(t, 'item'):
        return t.item()
    else:
        return t[0]

def reduce_tensor(tensor, world_size):
    rt = tensor.clone()