fix tqdm usage (facebookresearch#246)

Ricardozzf · Dec 5, 2018 · 46dbf29 · 46dbf29
1 parent 2009ed5
commit 46dbf29
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/maskrcnn_benchmark/engine/inference.py b/maskrcnn_benchmark/engine/inference.py
@@ -17,7 +17,7 @@ def compute_on_dataset(model, data_loader, device):
     model.eval()
     results_dict = {}
     cpu_device = torch.device("cpu")
-    for i, batch in tqdm(enumerate(data_loader)):
+    for i, batch in enumerate(tqdm(data_loader)):
         images, targets, image_ids = batch
         images = images.to(device)
         with torch.no_grad():