ColossalAI/colossalai/auto_parallel/checkpoint
Super Daniel 8ea50d999e
[hotfix] pass a parameter. (#2288)
* [autockpt] make it work.

* [autockpt] linearize / merge shape-consistency nodes.

* [autockpt] considering parameter and optimizer weights.

* [hotfix] pass a parameter.
2023-01-03 18:05:06 +08:00
..
__init__.py [autoparallel] move ckpt solvers to autoparallel folder / refactor code (#1764) 2022-11-01 10:43:15 +08:00
build_c_ext.py [autoparallel] refactor and add rotorc. (#1789) 2022-11-03 12:32:51 +08:00
ckpt_solver_base.py [autockpt] considering parameter and optimizer weights. (#2279) 2023-01-03 16:55:49 +08:00
ckpt_solver_chen.py [autockpt] considering parameter and optimizer weights. (#2279) 2023-01-03 16:55:49 +08:00
ckpt_solver_rotor.c [autoparallel] refactor and add rotorc. (#1789) 2022-11-03 12:32:51 +08:00
ckpt_solver_rotor.py [hotfix] pass a parameter. (#2288) 2023-01-03 18:05:06 +08:00
operation.py [autoparallel] refactor and add rotorc. (#1789) 2022-11-03 12:32:51 +08:00