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

3.0 KiB

Raw Blame History

View Raw