George Hotz c162e748f5 fix float64 warning on training

2021-10-30 20:07:31 -07:00

..

efficientnet

feat: add efficientnet test (#285)

2021-10-30 15:53:51 -07:00

__init__.py

All devices are equal! (#196)

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

test_efficientnet.py

not forcing 3.9 for a stupid type

2021-10-30 16:52:40 -07:00

test_gc.py

Test split (#231)

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

test_mnist.py

fix float64 warning on training

2021-10-30 20:07:31 -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

Cherry backprop for conv2d (#281)

2021-10-30 16:12:19 -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

only resnet18, it's too slow otherwise

2021-10-30 16:48:39 -07:00