mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-09-02 09:38:05 +00:00
[Gemini] ZeROHookV2 -> GeminiZeROHook (#1972)
This commit is contained in:
@@ -9,4 +9,4 @@ colossalai.zero.utils
|
||||
:maxdepth: 2
|
||||
|
||||
colossalai.zero.utils.zero_hook
|
||||
colossalai.zero.utils.zero_hook_v2
|
||||
colossalai.zero.utils.gemini_hook
|
||||
|
@@ -1,5 +1,5 @@
|
||||
colossalai.zero.utils.zero\_hook\_v2
|
||||
====================================
|
||||
|
||||
.. automodule:: colossalai.zero.utils.zero_hook_v2
|
||||
.. automodule:: colossalai.zero.utils.gemini_hook
|
||||
:members:
|
||||
|
Reference in New Issue
Block a user