Commit 46dbf29b authored by CoinCheung's avatar CoinCheung Committed by Francisco Massa

fix tqdm usage (#246)

parent 2009ed5e
......@@ -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():
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment