Frank Lee 250be4d31e [utils] integrated colotensor with lazy init context (#1324) ...

* [utils] integrated colotensor with lazy init context

* polish code

* polish code

* polish code

2022-07-15 17:47:12 +08:00

..

__init__.py

fix import error (#880)

2022-04-26 19:28:40 +08:00

colo_init_context.py

[tensor] distributed checkpointing for parameters (#1240)

2022-07-12 15:51:06 +08:00

lazy_init_context.py

[utils] integrated colotensor with lazy init context (#1324)

2022-07-15 17:47:12 +08:00

utils.py

[context]use meta tensor to init model lazily. (#1187)

2022-06-29 21:02:30 +08:00