You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/tests/kit/model_zoo/custom
Edenzzzz 43995ee436
[Feature] Distributed optimizers: Lamb, Galore, CAME and Adafactor (#5694)
7 months ago
..
__init__.py [Feature] Distributed optimizers: Lamb, Galore, CAME and Adafactor (#5694) 7 months ago
base.py [test] merge old components to test to model zoo (#4945) 1 year ago
hanging_param_model.py [test] merge old components to test to model zoo (#4945) 1 year ago
nested_model.py [test] merge old components to test to model zoo (#4945) 1 year ago
repeated_computed_layers.py [test] merge old components to test to model zoo (#4945) 1 year ago
simple_mlp.py [Feature] Distributed optimizers: Lamb, Galore, CAME and Adafactor (#5694) 7 months ago
simple_net.py [test] merge old components to test to model zoo (#4945) 1 year ago