Francis Lata 707099487a Multiprocessing UNet3D dataloader (#4801) ...

* testing dataloader

* matching dataloader implementation for unet3d

* remove comments

* clean up dataloader

* add cookie and cleanup

* use shm_path when creating SharedMemory

* add support for testing resnet and unet3d dataloaders

* update dataset test to return preprocesed data directory in prep for dataloader testing

* pass preprocessed dataset directory properly

* update loader function for dataloader

* add shuffling on indices

* update shm name

* more cleanup for unet3d dataloader

* remove changes to tests

---------

Co-authored-by: chenyu <chenyu@fastmail.com>

2024-06-02 11:30:47 -04:00

..

conversation_data

Whisper + LLAMA + VITS (#2332)

2023-12-02 15:03:46 -08:00

llm.c

fix train_gpt2.py (#4771)

2024-05-29 12:01:34 -04:00

mlperf

Multiprocessing UNet3D dataloader (#4801)

2024-06-02 11:30:47 -04:00

openpilot

move memory into schedule (#4597)

2024-05-15 07:54:20 -07:00

rl

more beautiful_cartpole with exposed hparams

2024-01-07 17:41:09 -08:00

sovits_helpers

move dtypes to dtype.py (#2964)

2024-01-01 14:58:48 -08:00

vgg7_helpers

move to new cached fetch (#2493)

2023-11-28 17:36:55 -08:00

webgl/yolov8

webgl backend in extra (#3041)

2024-01-08 09:29:13 -08:00

webgpu/stable_diffusion

s/lazydata.realized/lazydata.base.realized/g (#2914)

2023-12-22 14:45:13 -05:00

__init__.py

failing llama test

2023-03-11 16:28:10 -08:00

beautiful_cartpole.py

more beautiful_cartpole with exposed hparams

2024-01-07 17:41:09 -08:00

beautiful_mnist_multigpu.py

add beautiful_mnist_multigpu example

2024-04-02 00:54:04 +00:00

beautiful_mnist.py

hotfix: 100% accuracy is wrong

2024-05-01 08:07:18 -07:00

coder.py

apply the same fix_bf16 in llama and coder (#3789)

2024-03-17 21:25:24 -04:00

compile_efficientnet.py

webgl backend in extra (#3041)

2024-01-08 09:29:13 -08:00

compile_tensorflow.py

fix various examples (#4691)

2024-05-22 20:43:21 -04:00

conversation.py

fix conversation.py quantize (#4663)

2024-05-20 17:36:37 -04:00

efficientnet.py

remove clang program header (#4422)

2024-05-04 08:38:01 -07:00

gpt2.py

change Tensor.stack to method (#4719)

2024-05-24 17:04:19 -04:00

handcode_resnet50_opt.py

move graph/search to engine (#4596)

2024-05-14 23:12:59 -07:00

hlb_cifar10.py

move image into tensor.py. delete features (#4603)

2024-05-15 10:50:25 -07:00

index.html

Enable Multi-Output Export (#2179)

2023-10-30 18:42:26 -07:00

llama3.py

greater chat api endpoint compat (#4792)

2024-05-30 22:47:31 -07:00

llama.py

fix llama example quantize (#4699)

2024-05-23 15:35:26 -04:00

mamba.py

change Tensor.stack to method (#4719)

2024-05-24 17:04:19 -04:00

mask_rcnn.py

change Tensor.stack to method (#4719)

2024-05-24 17:04:19 -04:00

mixtral.py

remove mixtral weight to clang first (#3792)

2024-03-17 23:33:17 -04:00

mnist_gan.py

docs: showcase remove mnist_gan and add conversation.py (#4757)

2024-05-28 11:09:26 -04:00

serious_mnist.py

move state to nn/state (#1619)

2023-08-22 07:36:24 -07:00

simple_conv_bn.py

fix various examples (#4691)

2024-05-22 20:43:21 -04:00

so_vits_svc.py

change Tensor.stack to method (#4719)

2024-05-24 17:04:19 -04:00

stable_diffusion_seed0.png

validate stable diffusion for seed 0 (#2773)

2023-12-15 00:07:09 -05:00

stable_diffusion.py

minor cleanup example stable_diffusion (#4753)

2024-05-28 00:05:37 -04:00

train_efficientnet.py

move things, clean up extra (#2292)

2023-11-13 20:18:40 -08:00

train_resnet.py

move things, clean up extra (#2292)

2023-11-13 20:18:40 -08:00

transformer.py

fix onehot and jit in examples/transformer (#3073)

2024-01-10 02:22:41 -05:00

vgg7.py

waifu2x vgg7: testcase, auto-RGBA->RGB, function to grab pretrained models, training "fix" (#2117)

2023-10-19 22:07:15 -07:00

vit.py

move to new cached fetch (#2493)

2023-11-28 17:36:55 -08:00

vits.py

docs: showcase remove mnist_gan and add conversation.py (#4757)

2024-05-28 11:09:26 -04:00

whisper.py

remove the assign hack in whisper (#4240)

2024-04-20 23:56:44 -04:00

yolov3.py

Update yolov3.py (#2680)

2023-12-08 12:59:38 -08:00

yolov8-onnx.py

[ready] Replacing os with pathlib (#1708)

2023-08-30 10:41:08 -07:00

yolov8.py

change Tensor.stack to method (#4719)

2024-05-24 17:04:19 -04:00