Wenxuan Tan ec73f1b5e2 [CI] Cleanup Dist Optim tests with shared helper funcs (#6125) ...

* Refractor and cleanup using common helper funcs. Tests passed

* Update comments

* Fix relative import

* Fix param fetching bug

2025-02-12 13:42:34 +08:00

..

__init__.py

[Feature] Distributed optimizers: Lamb, Galore, CAME and Adafactor (#5694)

2024-05-14 13:52:45 +08:00

base.py

[test] merge old components to test to model zoo (#4945)

2023-10-20 10:35:08 +08:00

hanging_param_model.py

[test] merge old components to test to model zoo (#4945)

2023-10-20 10:35:08 +08:00

nested_model.py

[test] merge old components to test to model zoo (#4945)

2023-10-20 10:35:08 +08:00

repeated_computed_layers.py

[test] merge old components to test to model zoo (#4945)

2023-10-20 10:35:08 +08:00

simple_mlp.py

[CI] Cleanup Dist Optim tests with shared helper funcs (#6125)

2025-02-12 13:42:34 +08:00

simple_net.py

[test] merge old components to test to model zoo (#4945)

2023-10-20 10:35:08 +08:00