Unverified Commit eb84e5d5 authored by Erik's avatar Erik Committed by GitHub
Browse files

Update engine.py (#2187)

adding slight clarification to evaluation logic area, regarding images
parent a81d99b7
...@@ -81,13 +81,13 @@ def evaluate(model, data_loader, device): ...@@ -81,13 +81,13 @@ def evaluate(model, data_loader, device):
iou_types = _get_iou_types(model) iou_types = _get_iou_types(model)
coco_evaluator = CocoEvaluator(coco, iou_types) coco_evaluator = CocoEvaluator(coco, iou_types)
for image, targets in metric_logger.log_every(data_loader, 100, header): for images, targets in metric_logger.log_every(data_loader, 100, header):
image = list(img.to(device) for img in image) images = list(img.to(device) for img in images)
targets = [{k: v.to(device) for k, v in t.items()} for t in targets] targets = [{k: v.to(device) for k, v in t.items()} for t in targets]
torch.cuda.synchronize() torch.cuda.synchronize()
model_time = time.time() 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] outputs = [{k: v.to(cpu_device) for k, v in t.items()} for t in outputs]
model_time = time.time() - model_time model_time = time.time() - model_time
......
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