From c080fcdaab0a79ee15a45c39db40a6360407b638 Mon Sep 17 00:00:00 2001 From: Francis Lata Date: Thu, 5 Dec 2024 17:01:28 +0000 Subject: [PATCH] hotfix for dataloader typo causing some hang --- examples/mlperf/dataloader.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/mlperf/dataloader.py b/examples/mlperf/dataloader.py index 905330974c..e9df7f5ee2 100644 --- a/examples/mlperf/dataloader.py +++ b/examples/mlperf/dataloader.py @@ -380,7 +380,7 @@ def load_retinanet_data(base_dir:Path, val:bool, queue_in:Queue, queue_out:Queue match_quality_matrix = box_iou(tgt["boxes"], anchors) match_idxs = find_matches(match_quality_matrix, allow_low_quality_matches=True) match_idxs = np.clip(match_idxs, 0, None) - boxes, labels = tgt["boxes"][matches], tgt["labels"][matches] + boxes, labels = tgt["boxes"][match_idxs], tgt["labels"][match_idxs] Y_boxes[idx].contiguous().realize().lazydata.realized.as_buffer(force_zero_copy=True)[:] = boxes.tobytes() Y_labels[idx].contiguous().realize().lazydata.realized.as_buffer(force_zero_copy=True)[:] = labels.tobytes()