5 Commits

Author	SHA1	Message	Date
Hakjin Lee	46c009dba4	[format] Run lint on colossalai.engine (#3367)	2023-04-05 23:24:43 +08:00
Sze-qq	94eec1c5ad	[NFC] polish colossalai/engine/gradient_accumulation/_gradient_accumulation.py code style (#3277) ... Co-authored-by: siqi <siqi@siqis-MacBook-Pro.local>	2023-03-29 15:22:21 +08:00
Frank Lee	e4685832f8	[engine] fixed bug in gradient accumulation dataloader to keep the last step (#1030)	2022-05-26 14:28:23 +08:00
Frank Lee	11f54c7b6b	[doc] improved docstring and assertion messages for the engine module (#871)	2022-04-26 10:00:18 +08:00
Jiarui Fang	681addb512	[refactor] moving grad acc logic to engine (#804)	2022-04-19 14:03:21 +08:00