Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Model in training mode during demo.py #116

Open
braydenrudisill opened this issue Sep 9, 2024 · 1 comment
Open

Model in training mode during demo.py #116

braydenrudisill opened this issue Sep 9, 2024 · 1 comment

Comments

@braydenrudisill
Copy link

Model.eval() needs to be added in the demo.py for the model be in evaluation mode. Being stuck in training mode enables this line of code which uses the ground truth transformation.

# 7 Random select ground truth node correspondences during training
if self.training:
    ref_node_corr_indices, src_node_corr_indices, node_corr_scores = self.coarse_target(
        gt_node_corr_indices, gt_node_corr_overlaps
    )
@cduch9
Copy link

cduch9 commented Oct 18, 2024

Thanks a lot! It would have saved me a lot of time if I had seen this post earlier.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants