ColossalAI/colossalai/utils/model
Ziyue Jiang 7c530b9de2
[Tensor] add Parameter inheritance for ColoParameter (#1041)
* add Parameter inheritance for ColoParameter

* remove tricks

* remove tricks

* polish

* polish
2022-05-30 17:23:44 +08:00
..
__init__.py fix import error (#880) 2022-04-26 19:28:40 +08:00
colo_init_context.py [Tensor] add Parameter inheritance for ColoParameter (#1041) 2022-05-30 17:23:44 +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