This website requires JavaScript.
Explore
Help
Register
Sign In
github
/
ColossalAI
Watch
1
Star
0
Fork
0
You've already forked ColossalAI
mirror of
https://github.com/hpcaitech/ColossalAI.git
synced
2026-01-14 20:16:01 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
Files
54aabb8da461524a526302a1ec3b5d46b649feeb
ColossalAI
/
colossalai
/
engine
/
gradient_handler
History
Frank Lee
11f54c7b6b
[doc] improved docstring and assertion messages for the engine module (
#871
)
2022-04-26 10:00:18 +08:00
..
__init__.py
fix format (
#585
)
2022-04-06 11:40:59 +08:00
_base_gradient_handler.py
fix format (
#586
)
2022-04-06 11:40:59 +08:00
_data_parallel_gradient_handler.py
[doc] improved docstring and assertion messages for the engine module (
#871
)
2022-04-26 10:00:18 +08:00
_moe_gradient_handler.py
[doc] improved docstring and assertion messages for the engine module (
#871
)
2022-04-26 10:00:18 +08:00
_pipeline_parallel_gradient_handler.py
[doc] improved docstring and assertion messages for the engine module (
#871
)
2022-04-26 10:00:18 +08:00
_sequence_parallel_gradient_handler.py
[doc] improved docstring and assertion messages for the engine module (
#871
)
2022-04-26 10:00:18 +08:00
_zero_gradient_handler.py
[doc] improved docstring and assertion messages for the engine module (
#871
)
2022-04-26 10:00:18 +08:00
utils.py
add moe context, moe utilities and refactor gradient handler (
#455
)
2022-03-18 16:38:32 +08:00
