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