Commits · 8bcad7367769633699c4ec5b6d94f2119ff44a68 · OpenDAS / ColossalAI

10 May, 2023 1 commit

[booster] add tests for ddp and low level zero's checkpointio (#3715) · 20068ba1

jiangmingyan authored May 10, 2023

* [booster] update tests for booster

* [booster] update tests for booster

* [booster] update tests for booster

* [booster] update tests for booster

* [booster] update tests for booster

* [booster] update booster tutorials#3717, fix recursive check

20068ba1

05 May, 2023 1 commit

[booster] gemini plugin support shard checkpoint (#3610) · 307894f7

jiangmingyan authored May 05, 2023



* gemini plugin add shard checkpoint save/load

* gemini plugin add shard checkpoint save/load

* gemini plugin add shard checkpoint save/load

* gemini plugin add shard checkpoint save/load

* gemini plugin add shard checkpoint save/load

* gemini plugin add shard checkpoint save/load

* gemini plugin add shard checkpoint save/load

* gemini plugin add shard checkpoint save/load

* gemini plugin add shard checkpoint save/load

* gemini plugin add shard checkpoint save/load

* gemini plugin add shard checkpoint save/load

* gemini plugin add shard checkpoint save/load

* gemini plugin add shard checkpoint save/load

* gemini plugin add shard checkpoint save/load

* gemini plugin support shard checkpoint

* [API Refactoring]gemini plugin support shard checkpoint

* [API Refactoring]gemini plugin support shard checkpoint

* [API Refactoring]gemini plugin support shard checkpoint

* [API Refactoring]gemini plugin support shard checkpoint

* [API Refactoring]gemini plugin support shard checkpoint

* [API Refactoring]gemini plugin support shard checkpoint

* [API Refactoring]gemini plugin support shard checkpoint

* [API Refactoring]gemini plugin support shard checkpoint

* [API Refactoring]gemini plugin support shard checkpoint

* [API Refactoring]gemini plugin support shard checkpoint

* [API Refactoring]gemini plugin support shard checkpoint

* [API Refactoring]gemini plugin support shard checkpoint

* [API Refactoring]gemini plugin support shard checkpoint

---------
Co-authored-by: luchen <luchen@luchendeMBP.lan>
Co-authored-by: luchen <luchen@luchendeMacBook-Pro.local>

307894f7

06 Apr, 2023 2 commits

[checkpoint] support huggingface style sharded checkpoint (#3461) · 52a933e1

jiangmingyan authored Apr 06, 2023



* [checkpoint] support huggingface style sharded checkpoint

* [checkpoint] support huggingface style sharded checkpoint

* [checkpoint] support huggingface style sharded checkpoint

* [checkpoint] support huggingface style sharded checkpoint

* [checkpoint] support huggingface style sharded checkpoint

---------
Co-authored-by: luchen <luchen@luchendeMBP.lan>

52a933e1

[test] refactor tests with spawn (#3452) · 80eba05b

Frank Lee authored Apr 06, 2023

* [test] added spawn decorator

* polish code

* polish code

* polish code

* polish code

* polish code

* polish code

80eba05b

04 Apr, 2023 1 commit
- [checkpoint] refactored the API and added safetensors support (#3427) · 1beb85cc
  Frank Lee authored Apr 04, 2023
```
* [checkpoint] refactored the API and added safetensors support

* polish code
```
  1beb85cc
27 Mar, 2023 1 commit
- [booster] implemented the torch ddd + resnet example (#3232) · 73d3e4d3
  Frank Lee authored Mar 27, 2023
```
* [booster] implemented the torch ddd + resnet example

* polish code
```
  73d3e4d3
23 Mar, 2023 1 commit
- [api] implemented the checkpoint io module (#3205) · cd142fbe
  Frank Lee authored Mar 23, 2023
```
* [api] implemented the checkpoint io module

* polish code

* polish code
```
  cd142fbe