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