github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/tests/components_to_test/__init__.py

2 lines
83 B
Python
Raw Normal View History

[zero] adapt for no-leaf module in zero (#535) only process module's own parameters in Zero context add zero hooks for all modules that contrain parameters gather parameters only belonging to module itself
2022-03-28 09:42:18 +00:00
from . import repeated_computed_layer, resnet, nested_model, bert, no_leaf_module