Sohaib 65d09031f2 add retinanet with resnet backbone (#813) ...

* add retinanet with resnet backbone

* adds resnext to support loading retinanet pretrained on openimages
* object detection post processing with numpy
* data is downloaded and converted to coco format with fiftyone
* data loading and mAP evaluation with pycocotools

* remove fiftyone dep

* * eval freq

* fix model timing

* del jit for last batch

* faster accumulate

2023-05-28 20:20:16 -07:00

..

mlperf

add retinanet with resnet backbone (#813)

2023-05-28 20:20:16 -07:00

yolo

Permute examples (#731)

2023-03-29 05:07:06 +04: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_torch.py

hlb_cifar10 init from torch weights

2023-04-18 19:09:13 -07:00

hlb_cifar10.py

hlb_cifar10 init from torch weights

2023-04-18 19:09:13 -07:00

llama.py

Promote Embedding to nn (#798)

2023-05-25 18:39:45 -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

start on mlperf models

2023-05-10 16:30:49 -07:00

train_efficientnet.py

rename log_softmax, support dim, fix onnx Softmax

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

train_resnet.py

ResNet: match implementation with Nvidia and PyTorch (#770)

2023-05-10 09:01:22 -07: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

Permute examples (#731)

2023-03-29 05:07:06 +04:00

yolov8.py

yolov8 works up to the MaxPool

2023-02-22 19:32:13 -08:00