tinygrad/examples
Ahmed Harmouche 2114dc13d1
Allow multi-input model export (#1995)
* Allow multi-input model export

* Add model export unit test

* Fix efficientnet compilation

* Only run model export test on JIT supported devices

* Skip export model test if not EXPORT_SUPPORTED_DEVICE
2023-10-07 04:13:34 -07:00
..
mlperf better resnet eval (#1943) 2023-09-29 05:40:25 -07:00
sovits_helpers Implementation of SoftVC VITS SVC model (#1371) 2023-08-13 19:43:23 -07:00
vgg7_helpers [ready] Replacing os with pathlib (#1708) 2023-08-30 10:41:08 -07:00
__init__.py failing llama test 2023-03-11 16:28:10 -08:00
benchmark_train_efficientnet.py add cache collector (#1595) 2023-08-28 19:59:55 -07:00
compile_efficientnet.py Allow multi-input model export (#1995) 2023-10-07 04:13:34 -07:00
compile_tensorflow.py moved extras/jit.py -> tinygrad/jit.py (#599) 2023-02-25 08:32:33 -08:00
efficientnet.py Fix plt output comment (#1428) 2023-08-03 23:35:52 -07:00
gpt2.py simplify gpt2 example (#1973) 2023-10-05 07:09:29 -07:00
handcode_resnet50_opt.py no process() in Linearizer (#1966) 2023-10-04 07:18:42 -07:00
hlb_cifar10.py with Tensor.train() (#1935) 2023-09-28 18:02:31 -07:00
index.html simple exporting models (#1344) 2023-08-01 09:35:48 -07:00
llama.py Fix llama with empty prompt (#1997) 2023-10-06 06:48:07 -07:00
mask_rcnn.py MaskRCNN Inference (#884) 2023-06-25 15:37:51 -07:00
mnist_gan.py move state to nn/state (#1619) 2023-08-22 07:36:24 -07:00
serious_mnist.py move state to nn/state (#1619) 2023-08-22 07:36:24 -07:00
simple_conv_bn.py with Tensor.train() (#1935) 2023-09-28 18:02:31 -07:00
so_vits_svc.py use class Foo: instead of class Foo(): (#1797) 2023-09-06 12:20:25 -07:00
stable_diffusion.py schedule2, keep the tests working with small changes (#1932) 2023-09-28 09:14:43 -07:00
train_efficientnet.py Fix examples/train_efficientnet (#1947) 2023-10-02 02:23:38 -07:00
train_resnet.py move state to nn/state (#1619) 2023-08-22 07:36:24 -07:00
transformer.py move state to nn/state (#1619) 2023-08-22 07:36:24 -07:00
vgg7.py [ready] Replacing os with pathlib (#1708) 2023-08-30 10:41:08 -07:00
vit.py .cpu().numpy() -> .numpy() (#1594) 2023-08-21 09:53:29 -07:00
vits.py [ready] Replacing os with pathlib (#1708) 2023-08-30 10:41:08 -07:00
whisper.py [ready] Replacing os with pathlib (#1708) 2023-08-30 10:41:08 -07:00
yolov3.py .cpu().numpy() -> .numpy() (#1594) 2023-08-21 09:53:29 -07:00
yolov8-onnx.py [ready] Replacing os with pathlib (#1708) 2023-08-30 10:41:08 -07:00
yolov8.py use class Foo: instead of class Foo(): (#1797) 2023-09-06 12:20:25 -07:00