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

fix validation test case

Browse Source
This commit is contained in:
Francis Lata
2025-01-28 03:06:43 -08:00
parent fc957e7377
commit dcd1941b94
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/external/external_test_datasets.py vendored
View File

@@ -157,7 +157,7 @@ class TestOpenImagesDataset(ExternalTestDatasets):
ref_dataloader = self._create_ref_dataloader(base_dir, ann_file, "val")
transform = GeneralizedRCNNTransform(img_size, img_mean, img_std)
for ((tinygrad_img, _), (ref_img, _)) in zip(tinygrad_dataloader, ref_dataloader):
for ((tinygrad_img, _, _), (ref_img, _)) in zip(tinygrad_dataloader, ref_dataloader):
ref_img, _ = transform(ref_img.unsqueeze(0))
np.testing.assert_equal(tinygrad_img.numpy(), ref_img.tensors.transpose(1, 3).numpy())