You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai/auto_parallel/checkpoint
Boyuan Yao 8593ae1a3f
[autoparallel] rotor solver refactor (#2813)
2 years ago
..
__init__.py
build_c_ext.py
ckpt_solver_base.py [autockpt] considering parameter and optimizer weights. (#2279) 2 years ago
ckpt_solver_chen.py [autockpt] considering parameter and optimizer weights. (#2279) 2 years ago
ckpt_solver_rotor.c [autoparallel] rotor solver refactor (#2813) 2 years ago
ckpt_solver_rotor.py [autoparallel] rotor solver refactor (#2813) 2 years ago
operation.py