Automatic Mixed Precision Sum of different losses

[ptrblck]

You could use a single GradScaler and scale the final loss as described here.