From d69cba794eda4d67ce435f6ed7f8c83b75fb6f0d Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Wed, 27 Oct 2021 15:43:46 +0200 Subject: [PATCH] Update val.py --- val.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/val.py b/val.py index cd3c1b0326f0..abe9842f933c 100644 --- a/val.py +++ b/val.py @@ -160,8 +160,8 @@ def run(data, dt, p, r, f1, mp, mr, map50, map = [0.0, 0.0, 0.0], 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0 loss = torch.zeros(3, device=device) jdict, stats, ap, ap_class = [], [], [], [] - pbar = tqdm(enumerate(dataloader), desc=s, bar_format='{l_bar}{bar:10}{r_bar}{bar:-10b}') # progress bar - for batch_i, (img, targets, paths, shapes) in pbar: + pbar = tqdm(dataloader, desc=s, bar_format='{l_bar}{bar:10}{r_bar}{bar:-10b}') # progress bar + for batch_i, (img, targets, paths, shapes) in enumerate(pbar): t1 = time_sync() img = img.to(device, non_blocking=True) img = img.half() if half else img.float() # uint8 to fp16/32