Liam ebd72ff437 Test split (#231) ...

* Split tests

Split tests into "Test CPU" and "Test GPU".

Add test flag "TEST_DEVICES" which is a comma separated list of devices:
CPU,GPU,ANE

* Run tests based on provided TEST_DEVICES flag

By default will run all "CPU,GPU,ANE"

* fix bad quote

* Revert changes and use GPU=1

This is done through setting the default Tensor Device to Device.CPU of
GPU=1 is set.

Run GPU tests: GPU=1 pytest -s -v

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

..

augment.py

Big MNIST model with PIL augmentation and load/save (#160)

2020-12-13 20:45:55 -08:00

efficientnet.py

issue-193 - Move torch loader out of efficientnet code (#213)

2020-12-22 00:19:16 -05:00

gradcheck.py

move gradcheck to extra, clean up unbroadcast

2020-11-16 08:03:31 -08:00

training.py

Test split (#231)

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

transformer.py

if you like your transformers twice as slow, use the GPU

2020-12-29 17:14:23 -05:00

utils.py

issue-193 - Move torch loader out of efficientnet code (#213)

2020-12-22 00:19:16 -05:00