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

..

mnist

Improved __getitem__ (#254)

2021-05-05 22:15:22 -07:00

__init__.py

All devices are equal! (#196)

2020-12-15 23:44:08 -08:00

test_gc.py

Test split (#231)

2021-01-01 09:19:03 -05:00

test_mnist.py

Improved __getitem__ (#254)

2021-05-05 22:15:22 -07:00

test_net_speed.py

Test split (#231)

2021-01-01 09:19:03 -05:00

test_nn.py

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

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

test_ops.py

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

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

test_optim.py

Test split (#231)

2021-01-01 09:19:03 -05:00

test_tensor.py

Add dropout test (#265)

2021-06-19 08:49:13 -07:00

test_train.py

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

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