neodroidvision.utilities.torch_utilities.distributing.distributing_utilities.reduce_dict

neodroidvision.utilities.torch_utilities.distributing.distributing_utilities.reduce_dict(input_dict: dict, average: bool = True) dict[source]

Args: input_dict (dict): all the values will be reduced average (bool): whether to do average or sum Reduce the values in the dictionary from all processes so that all processes have the averaged results. Returns a dict with the same fields as input_dict, after reduction.