tinygrad/datasets
wozeparrot 2fd2fb6380
int8/uint8 support (#837)
* feat: int8 support

* feat: uint8 support

* feat: int8 tests

* fix: fix uint8 on clang

* feat: test casting between int8/uint8/float16/float32

* clean: way cleaner dtype tests

* feat: preprocess_imagenet using the correct dtype

* feat: add test for overflow between uint8 and int8
2023-05-28 23:15:06 -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
imagenet.py Cleanup mlperf (#797) 2023-05-25 11:36:43 -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