LuGY cbac782254 [zero]fix zero ckptIO with offload (#4529) ...

* fix zero ckptio with offload

* fix load device

* saved tensors in ckpt should be on CPU

* fix unit test

* fix unit test

* add clear cache

* save memory for CI

2023-09-01 17:41:19 +08:00

..

gemini

[gemini] improve compatibility and add static placement policy (#4479)

2023-08-24 09:29:25 +08:00

legacy

[nfc] fix typo colossalai/zero (#3923)

2023-06-08 00:01:29 +08:00

low_level

[zero]fix zero ckptIO with offload (#4529)

2023-09-01 17:41:19 +08:00

__init__.py

[gemini] improve compatibility and add static placement policy (#4479)

2023-08-24 09:29:25 +08:00

wrapper.py

[gemini] improve compatibility and add static placement policy (#4479)

2023-08-24 09:29:25 +08:00