1f22feb7dae28cbaa157e900f3444680a6fe258f,catalyst/utils/distributed.py,,distributed_mean,#Any#,54

Before Change

def distributed_mean(value: float):
    if torch.distributed.is_initialized():
        value = torch.tensor(
            value,
            dtype=torch.float,

After Change

    
    Computes distributed mean among all nodes
    
    if is_torch_distributed_initialized():
        value = torch.tensor(
            value,
            dtype=torch.float,

Instances