From 57d47189c32700f69398250e3bbfb586cc0d9c4a Mon Sep 17 00:00:00 2001 From: tchaton Date: Wed, 23 Jun 2021 18:46:34 +0100 Subject: [PATCH] remove addition --- torchmetrics/metric.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchmetrics/metric.py b/torchmetrics/metric.py index e250608bfd5..b26799ef0aa 100644 --- a/torchmetrics/metric.py +++ b/torchmetrics/metric.py @@ -214,7 +214,7 @@ def _sync_dist(self, dist_sync_fn: Callable = gather_all_tensors, process_group: for attr, reduction_fn in self._reductions.items(): # pre-processing ops (stack or flatten for inputs) - if isinstance(output_dict[attr], Sequence) and isinstance(output_dict[attr][0], Tensor): + if isinstance(output_dict[attr][0], Tensor): output_dict[attr] = torch.stack(output_dict[attr]) elif isinstance(output_dict[attr][0], list): output_dict[attr] = _flatten(output_dict[attr])