tinygrad/datasets
Daniel Hipke b4ce23e4b8
Make cross_process use cloudpickle (#1118)
* fix syntax issues in imagenet_download.py

* use cloudpickle in cross_process to make it work in Python 3.9+

* add cross_process test

* prevent unpickling on every function call

* add cloudpickle to setup.py

* add support for args/kwargs
2023-07-04 00:47:34 -07:00
..
mnist move datasets into datasets 2021-10-30 19:55:50 -07:00
__init__.py start on mlperf models 2023-05-10 16:30:49 -07:00
coco.py Create COCO data directory if it doesn't exist. (#1114) 2023-07-03 18:15:53 -07:00
imagenet.py imagenet loader minor cleanups 2023-06-28 05:08:09 +00:00
imagenet_download.py Make cross_process use cloudpickle (#1118) 2023-07-04 00:47:34 -07:00
kits19.py Add MLPerf UNet3D model (#775) 2023-05-28 20:38:19 -07:00
librispeech.py Cleanup mlperf (#797) 2023-05-25 11:36:43 -07:00
openimages.py add retinanet with resnet backbone (#813) 2023-05-28 20:20:16 -07:00
preprocess_imagenet.py int8/uint8 support (#837) 2023-05-28 23:15:06 -07:00
squad.py Add mlperf bert model (#803) 2023-05-27 14:53:32 -07:00