github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

hotfix for dataloader typo causing some hang

Browse Source
This commit is contained in:
Francis Lata
2024-12-05 17:01:28 +00:00
parent bb6f6075cd
commit c080fcdaab
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/mlperf/dataloader.py
View File

@@ -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()