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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

'phase' not used in cm_per_epoch()

iagooteroc opened this issue · comments

The "phase" parameter in the cm_per_epoch() function is not passed to the dice_confusion_matrix() function. Instead, the "train" mode is used even in the validation phase.

_, cm = eu.dice_confusion_matrix(output, correct_labels, self.num_class, mode='train')

Good catch! I updated the line. Thanks.