github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/engine/gradient_accumulation
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 [refactor] moving grad acc logic to engine (#804) 2022-04-19 14:03:21 +08:00
_gradient_accumulation.py [doc] improved docstring and assertion messages for the engine module (#871) 2022-04-26 10:00:18 +08:00