github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-11 18:14:16 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
46c9ba33daa1d8c044da1aaa00a67196d332e8f4
ColossalAI/tests/components_to_test/no_leaf_module.py
HELSON a30e2b4c24 [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 17:42:18 +08:00

1.3 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink