9 Commits

Author	SHA1	Message	Date
chenyu	42b1c9625b	skip test TestKiTS19Dataset::test_training_set (#10936) ... flaky	2025-06-23 14:27:24 -04:00
George Hotz	81b9c04574	move high level stuff to unit tests [pr] (#10708) ... * move high level stuff to unit tests [pr] * process replay on unit tests * fix pr, less compute * set omp num threads * set 200MB buffer size limit * delete junk * fix tests * faster * move test_indexing to unit * faster	2025-06-08 14:05:56 -07:00
Francis Lata	1a1087e3a0	cleanups on losses and dataset tests (#9538)	2025-03-21 17:03:18 -04:00
Francis Lata	eb95825eea	RetinaNet dataloader (#9442) ... * retinanet dataloader * remove batch_size from generate_anchors * refactor kits19 dataset tests * add tests for dataloader * fix testing setup and cleanups * remove unused import	2025-03-21 13:36:41 -04:00
chenyu	52e7003414	Revert "make kits19 dataset samples have small sizes (#8591)" (#8610) ... This reverts commit 76a03e950a.	2025-01-14 12:24:27 -05:00
Francis Lata	76a03e950a	make kits19 dataset samples have small sizes (#8591)	2025-01-14 08:27:45 -08:00
Francis Lata	ce61be16f1	clean up how preprocessed folder is defined (#5813)	2024-07-30 12:35:26 -04:00
Francis Lata	0345577032	UNet3D dataloader shared memory fix (#5465) ... * create separate SharedMemory between inputs and labels * update path check for shared mem * clean up unit test for dataset	2024-07-13 20:26:00 -04:00
Francis Lata	bb849a57d1	[MLPerf] UNet3D dataloader (#4343) ... * add support for train/val datasets for kits19 * split dataset into train and val sets * add tests for kits19 dataloader * add MLPerf dataset tests to CI * update unet3d model_eval script * fix linting * add nibabel * fix how mock dataset gets created * update ref implementation with permalink and no edits * clean up test and update rand_flip implementation * cleanups	2024-04-28 22:34:18 -04:00