Fix ProgressMeter usage in imagenet validation (pytorch#577)

pytorch/examples@2ea9f6a
mpc001 · Jun 20, 2019 · 1de2ff9 · 1de2ff9
1 parent 2ea9f6a
commit 1de2ff9
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/imagenet/main.py b/imagenet/main.py
@@ -306,8 +306,10 @@ def validate(val_loader, model, criterion, args):
     losses = AverageMeter('Loss', ':.4e')
     top1 = AverageMeter('Acc@1', ':6.2f')
     top5 = AverageMeter('Acc@5', ':6.2f')
-    progress = ProgressMeter(len(val_loader), batch_time, losses, top1, top5,
-                             prefix='Test: ')
+    progress = ProgressMeter(
+        len(val_loader),
+        [batch_time, losses, top1, top5],
+        prefix='Test: ')
 
     # switch to evaluate mode
     model.eval()