ed322ebaf136955b8f9f060d8fc49a6c252a0bba,allennlp/training/metrics/average.py,Average,__call__,#Average#Any#,24

Before Change

        
        _total_value = list(self.detach_tensors(value))[0]
        _count = 1
        if is_distributed():
            device = torch.device("cuda" if dist.get_backend() == "nccl" else "cpu")
            count = torch.tensor(_count, device=device)
            total_value = torch.tensor(_total_value, device=device)
            dist.all_reduce(count, op=dist.ReduceOp.SUM)
            dist.all_reduce(total_value, op=dist.ReduceOp.SUM)
            _count = count.item()
            _total_value = total_value.item()
        self._count += _count
        self._total_value += _total_value

    @overrides

After Change

            The value to average.
        
        self._count += dist_reduce(1, ReduceOp.SUM)
        self._total_value += dist_reduce(float(list(self.detach_tensors(value))[0]), ReduceOp.SUM)

    @overrides
    def get_metric(self, reset: bool = False):

Instances