Fix

potterhsu · Sep 6, 2018 · 7a7ffc2 · 7a7ffc2
1 parent 9dcfa03
commit 7a7ffc2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/evaluator.py b/evaluator.py
@@ -19,7 +19,7 @@ def __init__(self, dataset, path_to_data_dir, path_to_results_dir):
     def evaluate(self, model):
         all_image_ids, all_pred_bboxes, all_pred_labels, all_pred_probs = [], [], [], []
 
-        for batch_index, (image_id_batch, image_batch, scale_batch, _, _) in tqdm(enumerate(self.dataloader)):
+        for batch_index, (image_id_batch, image_batch, scale_batch, _, _) in enumerate(tqdm(self.dataloader)):
             image_id = image_id_batch[0]
             image = image_batch[0].cuda()
             scale = scale_batch[0]