Skip to content

Commit

Permalink
Update engine.py (#2187)
Browse files Browse the repository at this point in the history
adding slight clarification to evaluation logic area, regarding images
  • Loading branch information
Erik committed May 11, 2020
1 parent a81d99b commit eb84e5d
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions references/detection/engine.py
Original file line number Diff line number Diff line change
Expand Up @@ -81,13 +81,13 @@ def evaluate(model, data_loader, device):
iou_types = _get_iou_types(model)
coco_evaluator = CocoEvaluator(coco, iou_types)

for image, targets in metric_logger.log_every(data_loader, 100, header):
image = list(img.to(device) for img in image)
for images, targets in metric_logger.log_every(data_loader, 100, header):
images = list(img.to(device) for img in images)
targets = [{k: v.to(device) for k, v in t.items()} for t in targets]

torch.cuda.synchronize()
model_time = time.time()
outputs = model(image)
outputs = model(images)

outputs = [{k: v.to(cpu_device) for k, v in t.items()} for t in outputs]
model_time = time.time() - model_time
Expand Down

0 comments on commit eb84e5d

Please sign in to comment.