fix: ConfLoss returns Python int 0 for empty valid masks

[andrewwhitecdw]

Problem

fix: ConfLoss returns Python int 0 for empty valid masks

Fix

Replace:

        conf_loss1 = conf_loss1.mean() if conf_loss1.numel() > 0 else 0
        conf_loss2 = conf_loss2.mean() if conf_loss2.numel() > 0 else 0

with:

        conf_loss1 = conf_loss1.mean() if conf_loss1.numel() > 0 else conf_loss1.new_zeros(())
        conf_loss2 = conf_loss2.mean() if conf_loss2.numel() > 0 else conf_loss2.new_zeros(())

Files changed

• dust3r/losses.py