Jacky LeeandGitHub 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_torch.py

rename log_softmax, support dim, fix onnx Softmax

2023-02-24 10:11:24 -08:00

hlb_cifar10.py

moved extras/jit.py -> tinygrad/jit.py (#599)

2023-02-25 08:32:33 -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