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