issues
search
ai-med
/
quickNAT_pytorch
PyTorch Implementation of QuickNAT and Bayesian QuickNAT, a fast brain MRI segmentation framework with segmentation Quality control using structure-wise uncertainty
MIT License
101
stars
36
forks
source link
07012021: saving model on best val dice score
#29
Closed
jyotirmay123
closed
3 years ago
jyotirmay123
commented
3 years ago
Evaluating DS on the entire validation set, not on 10 sample selection. (previous setup)
Modified best_ds_mean saving and loading with checkpoints.
Saving model based on best ds mean and corresponding epoch model.