mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-10-26 11:05:02 +00:00
a30e2b4c24
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
2 lines
83 B
Python
2 lines
83 B
Python
from . import repeated_computed_layer, resnet, nested_model, bert, no_leaf_module
|