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.