tinygrad/examples
Fernando Vidal 73bd0b217b
add int64 as supported dtype from numpy (#699)
* add int64 as supported dtype from numpy

Without this, examples/transformer.py didn't run. With this change it runs successfully.

* Update helpers.py

* Update transformer.py

* Update training.py
2023-03-18 17:15:04 -07:00
..
yolo Cleanup yolo and remove stateless classes (#604) 2023-02-26 16:55:21 -08: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 moved extras/jit.py -> tinygrad/jit.py (#599) 2023-02-25 08:32:33 -08:00
hlb_cifar10_torch.py rename log_softmax, support dim, fix onnx Softmax 2023-02-24 10:11:24 -08:00
llama.py Fix llama 13B RAM usage (#710) 2023-03-18 13:50:09 -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 Use pathlib (#711) 2023-03-18 13:49:21 -07:00
train_efficientnet.py rename log_softmax, support dim, fix onnx Softmax 2023-02-24 10:11:24 -08:00
train_resnet.py Import get_parameters from tinygrad.nn (#559) 2023-02-17 15:22:26 -08: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 fix yolo webcam 2023-02-26 17:24:05 -08:00
yolov8.py yolov8 works up to the MaxPool 2023-02-22 19:32:13 -08:00