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.

Sign up for GitHub

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

Jump to bottom

Change `checkpoint_latest` to `checkpoint_best` in inference script to match the priority in SCT #78

Open

yw7 opened this issue Oct 28, 2024 · 0 comments

Collaborator

yw7 commented Oct 28, 2024 •

edited

Loading

Here we have to change checkpoint_latest to checkpoint_best to match the priority in SCT:

totalspineseg/totalspineseg/inference.py

Line 301 in 7d92550

    
           checkpoint = 'checkpoint_final.pth' if (nnUNet_results / step1_dataset / f'{nnUNetTrainer}__{nnUNetPlans}__{configuration}' / f'fold_{fold}' / 'checkpoint_final.pth').is_file() else 'checkpoint_latest.pth'

totalspineseg/totalspineseg/inference.py

Line 552 in 7d92550

    
           checkpoint = 'checkpoint_final.pth' if (nnUNet_results / step2_dataset / f'{nnUNetTrainer}__{nnUNetPlans}__{configuration}' / f'fold_{fold}' / 'checkpoint_final.pth').is_file() else 'checkpoint_latest.pth'

The text was updated successfully, but these errors were encountered:

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment