Remove comment

pytorch · Sep 4, 2024 · 9cc00a7 · 9cc00a7
1 parent 7c66841
commit 9cc00a7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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[
             [