Frank Lee e4685832f8 [engine] fixed bug in gradient accumulation dataloader to keep the last step (#1030)

2022-05-26 14:28:23 +08:00

..

gradient_accumulation

[engine] fixed bug in gradient accumulation dataloader to keep the last step (#1030)

2022-05-26 14:28:23 +08:00

gradient_handler

[doc] improved docstring and assertion messages for the engine module (#871)

2022-04-26 10:00:18 +08:00

ophooks

[refactor] moving memtracer to gemini (#801)

2022-04-19 10:13:08 +08:00

paramhooks

[doc] improved docstring and assertion messages for the engine module (#871)

2022-04-26 10:00:18 +08:00

schedule

[pipelinable]use pipelinable to support GPT model. (#903)

2022-05-11 09:23:58 +08:00

__init__.py

fix format (#580)

2022-04-06 11:40:59 +08:00

_base_engine.py

[doc] improved docstring and assertion messages for the engine module (#871)

2022-04-26 10:00:18 +08:00