提交 46dbf29b 编写于 作者: C CoinCheung 提交者: Francisco Massa

fix tqdm usage (#246)

上级 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% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册