lars optimizer + tests (#3631)

* lars optimizer + tests

* fix skip list!

* use id to compare in skip list

* go back to using set

* Tensor(bool) * Tensor(bool) is and

* don't lint external/mlperf_resnet

* whitespace

* add external_test_optim to opencl tests

* give mlperf task a name

* mlperf under onnx

* remove track_gnorm

* contiguous instead of realize

* assert momentum and weight decay positive

---------

Co-authored-by: chenyu <chenyu@fastmail.com>
This commit is contained in:
David Hou
2024-03-06 15:11:01 -08:00
committed by GitHub
parent b2e92d44fa
commit 0afaf70d57
6 changed files with 295 additions and 3 deletions

View File

@@ -32,4 +32,5 @@ exclude = [
"extra/",
"openpilot/",
"tinygrad/runtime/autogen",
"test/external/mlperf_resnet",
]