Unverified Commit f5bf9d52 authored by dmitrysarov's avatar dmitrysarov Committed by GitHub
Browse files

remove reduntant target send to device (#2503)


Co-authored-by: default avatardmitrysarov <d.shaulskiy@gmail.com>
parent 37799adc
......@@ -83,7 +83,6 @@ def evaluate(model, data_loader, device):
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()
......
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