Repeated Evaluation in ImageNet

[Doyup Lee]

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.

[Norman Mu]

Fixed. Thanks!