facebookresearch / mae

PyTorch implementation of MAE https//arxiv.org/abs/2111.06377
Other
7.2k stars 1.2k forks source link

Why not use RandomSampler instead of SequentialSampler for validation set? #139

Open ucalyptus2 opened 1 year ago

ucalyptus2 commented 1 year ago

Hi @endernewton, @KaimingHe great paper.

netw0rkf10w commented 1 year ago

@forkbabu Why do you need the validation to be random?

ucalyptus2 commented 1 year ago

@netw0rkf10w because AUC score is 0 if the minibatch has only one type of labels. if labels are 0 and 1 and my minibatch has all 1s, the AUC score is 0.

netw0rkf10w commented 1 year ago

@forkbabu Why do you need to compute AUC for each minibatch?

ucalyptus2 commented 1 year ago

@netw0rkf10w just need the running mean of auc from each minibatch as my final auc after every validation epoch image

netw0rkf10w commented 1 year ago

I guess you should keep counting (using accumulating sums) the numbers of true positives and false positives, etc. and only compute AUC at the end.

ucalyptus2 commented 1 year ago

@netw0rkf10w ok thanks