ColossalAI/colossalai/utils/model
Ziyue Jiang d73c2b1d79
[Tensor] fix init context (#931)
* change torch.Parameter to ColoParameter

* fix post assignment for init context

* polish

* polish
2022-05-11 15:48:12 +08:00
..
__init__.py fix import error (#880) 2022-04-26 19:28:40 +08:00
colo_init_context.py [Tensor] fix init context (#931) 2022-05-11 15:48:12 +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