trainning EfficientLoFTR

[AlmogHagag]

Hi,
when i try to train, i get an error in function _compute_local_loss_l2 file loftr_loss.py :
on line:
offset_l2 = ((expec_f_gt[correct_mask] - expec_f[correct_mask]) ** 2).sum(-1)

data['expec_f'] is calculated couple of lines above :

[AlmogHagag]

maybe we need to del data['expec_f'] at the start of function?