tinygrad/examples
Jacky Lee 5d212864b5
Add MLPerf UNet3D model (#775)
* Add ResNet inference test and cannon

* Test with ResNet50

* test_car works with resnet fix

* Add KiTS19 dataset

* KiTS19: Implement iterate

* No batch load for this dataset

* Save results on iterate

* Implement dice score

* Add data prep and eval functions

* Resolve shape issue

* Conversion works but wrong values

* Segfaults when load_from_pretrained is called

* Fix segfault and assign properly

* Final result generated, though very slow

* Store and load final result to save time

* Fix typo in finalize

* Score computes

* More bug fixes, dice score is very low

* Working broken code

* Assign output values to result

* Getting a much higher score now

* Fix dataset preprocessing

* Mean DICE score of 88.5

* Ugh, typo

* Attempt to reimplement model

* Rename layers

* Tiny model works, kinda

* Accuracy? gone

* Implement InstanceNorm and match torch

* Test instance norm 2d and 3d

* Combined input block with downsample block

* Tiny model works, support strided convtranspose

* Commands to download dataset

* Clean up a bit

* unet3d_v2 -> unet3d

* Remove duplicated code

* Oops, put tests back
2023-05-28 20:38:19 -07:00
..
mlperf Add MLPerf UNet3D model (#775) 2023-05-28 20:38:19 -07:00
yolo Permute examples (#731) 2023-03-29 05:07:06 +04:00
__init__.py failing llama test 2023-03-11 16:28:10 -08:00
benchmark_train_efficientnet.py rename log_softmax, support dim, fix onnx Softmax 2023-02-24 10:11:24 -08:00
compile_efficientnet.py Devicebufferless (#708) 2023-03-18 14:40:23 -07:00
compile_tensorflow.py moved extras/jit.py -> tinygrad/jit.py (#599) 2023-02-25 08:32:33 -08:00
deep_deterministic_policy_gradient.py Fixed DDPG example (#667) 2023-03-09 11:49:52 -08:00
efficientnet.py Cast image to float32 (#702) 2023-03-14 08:13:19 -07:00
hlb_cifar10.py hlb_cifar10 init from torch weights 2023-04-18 19:09:13 -07:00
hlb_cifar10_torch.py hlb_cifar10 init from torch weights 2023-04-18 19:09:13 -07:00
llama.py Promote Embedding to nn (#798) 2023-05-25 18:39:45 -07:00
mnist_gan.py Cast Tensor data to float32 (#703) 2023-03-14 23:09:41 -07:00
serious_mnist.py rename log_softmax, support dim, fix onnx Softmax 2023-02-24 10:11:24 -08:00
stable_diffusion.py start on mlperf models 2023-05-10 16:30:49 -07:00
train_efficientnet.py rename log_softmax, support dim, fix onnx Softmax 2023-02-24 10:11:24 -08:00
train_resnet.py ResNet: match implementation with Nvidia and PyTorch (#770) 2023-05-10 09:01:22 -07:00
transformer.py add int64 as supported dtype from numpy (#699) 2023-03-18 17:15:04 -07:00
vgg7.py Remove Tensor.data (#565) 2023-02-18 16:36:12 -08:00
vit.py Remove Tensor.data (#565) 2023-02-18 16:36:12 -08:00
yolov3.py Permute examples (#731) 2023-03-29 05:07:06 +04:00
yolov8.py yolov8 works up to the MaxPool 2023-02-22 19:32:13 -08:00