Question for inliers score

[codeslake]

https://github.com/ignacio-rocco/weakalign/blob/52505eaeed52dc82cf755919b119ef99fe8b1ed1/train_weak.py#L126

Hi, First, thank you for sharing your work.

For the line of code attached, why are you passing "corr_aff" for the match? Shouldn't it be "corr_aff_tps"?

"mask_id" in the loss computation code goes through transformation up to tps. Wouldn't it be more appropriate to masking it with "corr_aff_tps"?

Thanks.