Yixiang Gao 094d3d71be with Tensor.train() (#1935) ...

* add with.train

* remove the rest TODOs

* fix pyflake

* fix pyflake error

* fix mypy

2023-09-28 18:02:31 -07:00

..

efficientnet

test cleanups

2023-03-03 06:36:06 -08:00

test_bert.py

move device to ops (#1646)

2023-08-23 08:30:17 -07:00

test_efficientnet.py

Inference test: add tests for ResNet50 (#773)

2023-05-13 21:18:15 -07:00

test_end2end.py

with Tensor.train() (#1935)

2023-09-28 18:02:31 -07:00

test_mnist.py

move state to nn/state (#1619)

2023-08-22 07:36:24 -07:00

test_onnx.py

update openpilot supercombo.onnx to 0.9.4 (#1681)

2023-08-26 19:16:08 -04:00

test_real_world.py

with Tensor.train() (#1935)

2023-09-28 18:02:31 -07:00

test_rnnt.py

.cpu().numpy() -> .numpy() (#1594)

2023-08-21 09:53:29 -07:00

test_train.py

move state to nn/state (#1619)

2023-08-22 07:36:24 -07:00