InnerEye-DeepLearning icon indicating copy to clipboard operation
InnerEye-DeepLearning copied to clipboard

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

AB#4142

ant0nsc avatar Jun 22 '21 13:06 ant0nsc