ColossalAI/colossalai/utils/model
Ziyue Jiang dfc88b85ea
[Tensor] simplify named param (#928)
* simplify ColoModulize

* simplify ColoModulize

* polish

* polish
2022-05-11 10:54:19 +08:00
..
__init__.py fix import error (#880) 2022-04-26 19:28:40 +08:00
colo_init_context.py [Tensor] simplify named param (#928) 2022-05-11 10:54:19 +08:00
pipelinable.py [pipelinable]use pipelinable to support GPT model. (#903) 2022-05-11 09:23:58 +08:00
utils.py [pipelinable]use pipelinable context to initialize non-pipeline model (#816) 2022-04-24 13:03:12 +08:00