Aggregate SoftDice loss for segmentation models over all GPUs
Open
ant0nsc
opened this issue 3 years ago
•
0 comments
Present behaviour: loss is computed per GPU
Try out if we can synchronize the tensors before computing the loss, so that it is computed off a larger effective batch size