google-research/augmix

Repeated Evaluation in ImageNet

LeeDoYup opened this issue · 1 comments

augmix/imagenet.py

Lines 485 to 489 in 7c84885

corruption_accs = test_c(net, test_transform)
for c in CORRUPTIONS:
print('\t'.join(map(str, [c] + corruption_accs[c])))
print('mCE (normalized by AlexNet):', compute_mce(corruption_accs))

in imagenet.py,
the evaluation of robustness is unnecessarily repeated in each iteration.

Fixed. Thanks!