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

…

base.py

…

hanging_param_model.py

…

nested_model.py

…

repeated_computed_layers.py

…

simple_mlp.py

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

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

simple_net.py

…