Torch DistributedContext repr, report size, not rank

returnn/torch/distributed.py

@@ -58,7 +58,7 @@ def __init__(self, options: Dict[str, Any]):
         self._check_no_unknown_opts()
 
     def __repr__(self):
-        return f"<{self.__class__.__name__} rank={self._rank} reduce_type={self._reduce_type}>"
+        return f"<{self.__class__.__name__} size={self._size} reduce_type={self._reduce_type}>"
 
     def _check_no_unknown_opts(self):
         # We check that all opts in self._opts have been used.