From 9cc00a778ac662b5df665962e71dffc843ff9db5 Mon Sep 17 00:00:00 2001 From: Sadra Barikbin Date: Wed, 4 Sep 2024 18:23:36 +0330 Subject: [PATCH] Remove comment --- ignite/distributed/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ignite/distributed/utils.py b/ignite/distributed/utils.py index 6c1357203c6..98c85d29b6f 100644 --- a/ignite/distributed/utils.py +++ b/ignite/distributed/utils.py @@ -371,7 +371,7 @@ def all_gather_tensors_with_shapes( padded_tensor = torch.nn.functional.pad( tensor, tuple(itertools.chain.from_iterable(map(lambda dim_size: (0, dim_size), reversed(padding_sizes)))) ) - all_padded_tensors: torch.Tensor = _model.all_gather(padded_tensor, group=group) # .split(max_shape[0], dim=0) + all_padded_tensors: torch.Tensor = _model.all_gather(padded_tensor, group=group) return [ all_padded_tensors[ [