ColossalAI/colossalai/utils/model
ver217 6ae8ed0407
[lazyinit] add correctness verification (#3147)
* [lazyinit] fix shared module

* [tests] add lazy init test utils

* [tests] add torchvision for lazy init

* [lazyinit] fix pre op fn

* [lazyinit] handle legacy constructor

* [tests] refactor lazy init test models

* [tests] refactor lazy init test utils

* [lazyinit] fix ops don't support meta

* [tests] lazy init test timm models

* [lazyinit] fix set data

* [lazyinit] handle apex layers

* [tests] lazy init test transformers models

* [tests] lazy init test torchaudio models

* [lazyinit] fix import path

* [tests] lazy init test torchrec models

* [tests] update torch version in CI

* [tests] revert torch version in CI

* [tests] skip lazy init test
2023-03-17 13:49:04 +08:00
..
__init__.py fix import error (#880) 2022-04-26 19:28:40 +08:00
colo_init_context.py [gemini] fix colo_init_context (#2683) 2023-02-13 17:53:15 +08:00
experimental.py [lazyinit] add correctness verification (#3147) 2023-03-17 13:49:04 +08:00
lazy_init_context.py [utils] fixed lazy init context (#1867) 2022-11-10 15:17:20 +08:00
utils.py [doc] add deepspeed citation and copyright (#2996) 2023-03-04 20:08:11 +08:00