Skip to content

F
fcos

Jeffcarl / fcos
与 Fork 源项目一致

从无法访问的项目Fork

通知 1
Star 0
Fork 0
F
fcos

前往新版Gitcode，体验更适合开发者的 AI 搜索 >>

提交 46dbf29b 编写于 作者: C CoinCheung 提交者： Francisco Massa
浏览文件
操作

fix tqdm usage (#246)

上级 2009ed5e
显示空白变更内容
内联 并排
Showing with 1 addition and 1 deletion
maskrcnn_benchmark/engine/inference.py
浏览文件 @ 46dbf29b
...@@ -17,7 +17,7 @@ def compute_on_dataset(model, data_loader, device): ...@@ -17,7 +17,7 @@ def compute_on_dataset(model, data_loader, device):
model.eval() model.eval()
results_dict = {} results_dict = {}
cpu_device = torch.device("cpu") 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, targets, image_ids = batch
images = images.to(device) images = images.to(device)
with torch.no_grad(): with torch.no_grad():
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或