mirror of https://github.com/hpcaitech/ColossalAI
660eed9124
* set optimizer to optional in execute_pipeline * arrange device and mixed precision in booster init * fix execute_pipeline in booster.py |
||
---|---|---|
.. | ||
__init__.py | ||
_utils.py | ||
base.py | ||
interleaved_pp.py | ||
one_f_one_b.py |