From 09f41c9689c4675b725982c668b4f014801d0f11 Mon Sep 17 00:00:00 2001 From: naelsondouglas Date: Tue, 2 Nov 2021 05:07:18 -0300 Subject: [PATCH] removed a pitfall where a built-in name was being shadowed --- sotabench.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/sotabench.py b/sotabench.py index 67816ff..dac1d76 100644 --- a/sotabench.py +++ b/sotabench.py @@ -55,10 +55,10 @@ def get_img_id(image_name): return image_name.split('/')[-1].replace('.JPEG', '') with torch.no_grad(): - for i, (input, target) in enumerate(test_loader): - input = input.to(device='cuda', non_blocking=True) + for i, (data, target) in enumerate(test_loader): + data = data.to(device='cuda', non_blocking=True) target = target.to(device='cuda', non_blocking=True) - output = model(input) + output = model(data) image_ids = [get_img_id(img[0]) for img in test_loader.dataset.imgs[i*test_loader.batch_size:(i+1)*test_loader.batch_size]] evaluator.add(dict(zip(image_ids, list(output.cpu().numpy())))) if evaluator.cache_exists: