ColossalAI/colossalai/utils/model
Ziyue Jiang df9dcbbff6
[Tensor] add hybrid device demo and fix bugs (#1059)
2022-06-03 12:09:49 +08:00
..
__init__.py fix import error (#880) 2022-04-26 19:28:40 +08:00
colo_init_context.py [Tensor] add hybrid device demo and fix bugs (#1059) 2022-06-03 12:09:49 +08:00
pipelinable.py [tensor] refactor colo-tensor (#992) 2022-05-19 12:44:59 +08:00
utils.py [pipelinable]use pipelinable context to initialize non-pipeline model (#816) 2022-04-24 13:03:12 +08:00