tinygrad

14,176 Commits 336 Branches 15 Tags

2 Commits

Include renames

Author	SHA1	Message	Date
Francis LataandGitHub	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
Francis Lata GitHub chenyu	3644077a42	[MLPerf][UNet3D] Add DICE loss + metrics (#4204 ) * add DICE loss and metrics * update dice to include reference implementation's link * remove unused imports * remove unnecessary test file and update pred + label for metrics and losses test * add tests to CI + add exclusion of mlperf_unet3d --------- Co-authored-by: chenyu <[email protected]>	2024-04-17 20:09:33 -04:00

Author

SHA1

Message

Date

Francis LataandGitHub

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

3644077a42

[MLPerf][UNet3D] Add DICE loss + metrics (#4204 )

* add DICE loss and metrics

* update dice to include reference implementation's link

* remove unused imports

* remove unnecessary test file and update pred + label for metrics and losses test

* add tests to CI + add exclusion of mlperf_unet3d

---------

Co-authored-by: chenyu <[email protected]>

2024-04-17 20:09:33 -04:00