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_download.py

Make cross_process use cloudpickle (#1118)

2023-07-04 00:47:34 -07:00

imagenet.py

imagenet loader minor cleanups

2023-06-28 05:08:09 +00: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