Guglielmo Camporese 2b7589db64 Added ResNet-{18, 34, 50, 101, 152} (#271) ...

* added resnets

* fix minor

* fix minor

* resnet in models

* added resnet test

* added resnet train test

* added linear, conv2d nn tests

* fix minor in extra/training

* resnet in models

* fix minor

* fix tolerance for linear in nn test

* fix eval, this causes cpu and gpu UT failing

* revert transformer test

* fix minor for CPU test

* improved model get_params for sequential layer

* fix minor for params counting

* commented broken ops tests

* improved train for resnet

2021-06-21 09:37:24 -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

minor optimizations & cleaning (#257)

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

yolov3.py

Add asserts for non-zero indices (#264)

2021-06-13 21:14:46 -07:00