neodroidvision.utilities.torch_utilities.distributing.metrics

Created on 23/03/2020

Functions

write_metrics_recursive(eval_result, prefix, ...)

param eval_result

reduce_loss_dict(loss_dict)

Reduce the loss dictionary from all processes so that process with rank 0 has the averaged results.