George Hotz 121d5a17ee use tinynn for Conv2d

2021-10-30 19:40:44 -07:00

..

benchmark.py

ane refactor, fix examples

2020-12-18 09:19:06 -08:00

efficientnet.py

move good models to models, add a training step test

2021-06-19 11:24:15 -07:00

mnist_gan.py

fixed example broken after gpu refactor (#238)

2021-01-05 07:41:54 -08:00

serious_mnist.py

minor optimizations & cleaning (#257)

2021-06-02 09:57:15 -07:00

train_efficientnet.py

fixed pylint, formatted python files iwth cblack on localhost (#204)

2020-12-17 14:37:31 -08:00

train_resnet.py

Added ResNet-{18, 34, 50, 101, 152} (#271)

2021-06-21 09:37:24 -07:00

transformer.py

move good models to models, add a training step test

2021-06-19 11:24:15 -07:00

use_ane.py

fixed pylint, formatted python files iwth cblack on localhost (#204)

2020-12-17 14:37:31 -08:00

vgg7.py

vgg7 (image upscaling) implementation - not the best, but it works (#255)

2021-05-12 23:48:51 -07:00

yolo_nn.py

use tinynn for Conv2d

2021-10-30 19:40:44 -07:00

yolov3.py

use tinynn for Conv2d

2021-10-30 19:40:44 -07:00