tinygrad/examples
Jacky Lee 0f58c4c648
Cleanup yolo and remove stateless classes (#604)
* Add AvgPool2d as a layer

* Clean up a bit

* Remove stateless layers in yolo_nn

* More cleanup

* Save label for test

* Add test for YOLO

* Test without cv2

* Don't fail if cv2 not installed

* Better import

* Fix image read

* Use opencv :)

* Don't download the file

* Fix errors

* Use same version

* Set higher confidence

* Why is the confidence so low?

* Start over

* Remove stateless layers

* Remove extra lines

* Revert changes

* Save a few more lines
2023-02-26 16:55:21 -08:00
..
yolo Cleanup yolo and remove stateless classes (#604) 2023-02-26 16:55:21 -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 moved extras/jit.py -> tinygrad/jit.py (#599) 2023-02-25 08:32:33 -08: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 Import get_parameters from tinygrad.nn (#559) 2023-02-17 15:22:26 -08:00
efficientnet.py moved extras/jit.py -> tinygrad/jit.py (#599) 2023-02-25 08:32:33 -08: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
mnist_gan.py rename log_softmax, support dim, fix onnx Softmax 2023-02-24 10:11:24 -08:00
serious_mnist.py rename log_softmax, support dim, fix onnx Softmax 2023-02-24 10:11:24 -08:00
stable_diffusion.py Replace phrase for prompt (#555) 2023-02-12 09:04:44 -08: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 Import get_parameters from tinygrad.nn (#559) 2023-02-17 15:22:26 -08: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 Cleanup yolo and remove stateless classes (#604) 2023-02-26 16:55:21 -08:00
yolov8.py yolov8 works up to the MaxPool 2023-02-22 19:32:13 -08:00