87a61ad92a9e0129e5c81c242f0ea96d77e6b0af,allennlp/training/metrics/attachment_scores.py,AttachmentScores,__call__,#AttachmentScores#Any#Any#Any#Any#Any#,36

Before Change

        self._total_words += correct_indices.numel() - (~mask).sum()

        if is_distributed():
            _total_sentences = torch.tensor(self._total_sentences).to(device)
            _total_words = torch.tensor(self._total_words).to(device)
            dist.all_reduce(_total_sentences, op=dist.ReduceOp.SUM)
            dist.all_reduce(_total_words, op=dist.ReduceOp.SUM)
            self._total_sentences = _total_sentences.item()
            self._total_words = _total_words.item()

    def get_metric(
        self, reset: bool = False, cuda_device: Union[int, torch.device] = torch.device("cpu"),

After Change

            dist.all_reduce(unlabeled_exact_match, op=dist.ReduceOp.SUM)
            dist.all_reduce(correct_labels_and_indices, op=dist.ReduceOp.SUM)
            dist.all_reduce(labeled_exact_match, op=dist.ReduceOp.SUM)
            total_sentences = torch.tensor(total_sentences).to(device)
            total_words = torch.tensor(total_words).to(device)
            dist.all_reduce(total_sentences, op=dist.ReduceOp.SUM)
            dist.all_reduce(total_words, op=dist.ReduceOp.SUM)
            total_sentences = total_sentences.item()
            total_words = total_words.item()

        self._unlabeled_correct += correct_indices.sum()
        self._exact_unlabeled_correct += unlabeled_exact_match.sum()

Instances