07012021: saving model on best val dice score

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.