neodroidvision.utilities.torch_utilities.distributing.distributing_utilities.all_gather_cuda

neodroidvision.utilities.torch_utilities.distributing.distributing_utilities.all_gather_cuda(data: Any) → List[bytes]

Run all_gather on arbitrary picklable data (not necessarily tensors) Args: data: any picklable object Returns: list[data]: list of data gathered from each rank