1
0
Fork 0
tinygrab/examples
George Hotz c7b38b324b
A beautiful MNIST training example (#2272)
* beautiful mnist

* beautiful mnist example

* from tinygrad import Tensor

* more beautiful

* the jit is super core tinygrad

* globalcounters reset on jit run

* symlinks and exclude

* beautiful_cartpole

* evaluate is it's own function

* no symlinks

* more beautiful

* jit reset for double speed

* type hinting for JIT

* beautiful_mnist gets 98%

* beautiful_mnist < 4s with BEAM=2

* better cartpole

* use actor critic

* zero_grad got lost

* delete double relu

* stable cartpole with PPO

* beautiful_cartpole is more beautiful

* REPLAY_BUFFER

* beautiful stuff typechecks

* None support in shape

* hp tuning
2023-11-17 19:42:43 -08:00
..
mlperf move things, clean up extra (#2292) 2023-11-13 20:18:40 -08:00
sovits_helpers Implementation of SoftVC VITS SVC model (#1371) 2023-08-13 19:43:23 -07:00
vgg7_helpers waifu2x vgg7: testcase, auto-RGBA->RGB, function to grab pretrained models, training "fix" (#2117) 2023-10-19 22:07:15 -07:00
webgpu/stable_diffusion Stable diffusion WebGPU port (#1370) 2023-11-03 18:29:16 -07:00
__init__.py failing llama test 2023-03-11 16:28:10 -08:00
beautiful_cartpole.py A beautiful MNIST training example (#2272) 2023-11-17 19:42:43 -08:00
beautiful_mnist.py A beautiful MNIST training example (#2272) 2023-11-17 19:42:43 -08:00
benchmark_train_efficientnet.py move things, clean up extra (#2292) 2023-11-13 20:18:40 -08:00
compile_efficientnet.py move things, clean up extra (#2292) 2023-11-13 20:18:40 -08:00
compile_tensorflow.py moved extras/jit.py -> tinygrad/jit.py (#599) 2023-02-25 08:32:33 -08:00
efficientnet.py move things, clean up extra (#2292) 2023-11-13 20:18:40 -08:00
f16_w_uint32.py add exp2 (#2192) 2023-10-31 17:48:42 -07:00
gpt2.py two stage cumsum in tensor.py (#2331) 2023-11-16 12:09:53 -08:00
handcode_resnet50_opt.py move things, clean up extra (#2292) 2023-11-13 20:18:40 -08:00
hlb_cifar10.py feat: remove cache_id (#2236) 2023-11-08 08:09:21 -08:00
index.html Enable Multi-Output Export (#2179) 2023-10-30 18:42:26 -07:00
llama.py Profiling-helper (#2321) 2023-11-16 14:15:56 -08:00
mask_rcnn.py move things, clean up extra (#2292) 2023-11-13 20:18:40 -08: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 clean up attentions in stable diffusion (#2275) 2023-11-11 14:25:36 -05: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 move things, clean up extra (#2292) 2023-11-13 20:18:40 -08: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 things, clean up extra (#2292) 2023-11-13 20:18:40 -08:00
vits.py [ready] Replacing os with pathlib (#1708) 2023-08-30 10:41:08 -07:00
whisper.py whisper: support batch inference, add librispeech WER test (#2074) 2023-11-16 13:50:08 -08: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