Commits · b09adff724c2bbded1c71cc51a707f736a0e2899 · OpenDAS / ColossalAI · GitLab

29 Jul, 2022 1 commit
- [hotfix] fix a running error in test_colo_checkpoint.py (#1387) · 527758b2
  HELSON authored Jul 29, 2022
  
  527758b2
19 Jul, 2022 1 commit
- [checkpoint] use gather_tensor in checkpoint and update its unit test (#1339) · f92c100d
  HELSON authored Jul 19, 2022
  
  f92c100d
15 Jul, 2022 1 commit
- [checkpoint] add ColoOptimizer checkpointing (#1316) · 9e4c6449
  Jiarui Fang authored Jul 15, 2022
  
  9e4c6449
14 Jul, 2022 3 commits
- [Optimizer] Remove useless ColoOptimizer (#1312) · 85f933b5
  Jiarui Fang authored Jul 14, 2022
  
  85f933b5
- [Optimizer] polish the init method of ColoOptimizer (#1310) · 9f105243
  Jiarui Fang authored Jul 14, 2022
  
  9f105243
- [checkpoint] add test for bert and hotfix save bugs (#1297) · 3ef3791a
  Jiarui Fang authored Jul 14, 2022
  
  3ef3791a
12 Jul, 2022 1 commit
- [tensor] distributed checkpointing for parameters (#1240) · c92f84fc
  Jiarui Fang authored Jul 12, 2022
  
  c92f84fc
11 Jul, 2022 1 commit
- [tensor] a shorter shard and replicate spec (#1245) · 9bcd2fd4
  Jiarui Fang authored Jul 11, 2022
  
  9bcd2fd4
08 Jul, 2022 2 commits
- [checkpoint] save sharded optimizer states (#1237) · 20da6e48
  Jiarui Fang authored Jul 08, 2022
  
  20da6e48
- [tensor] fix some unittests (#1234) · 3b500984
  Jiarui Fang authored Jul 08, 2022
  
  3b500984
07 Jul, 2022 1 commit
- [checkpoint]support generalized scheduler (#1222) · 04537bf8
  Yi Zhao authored Jul 07, 2022
  
  04537bf8
06 Jul, 2022 2 commits
- [checkpoint] make unitest faster (#1217) · 52736205
  Jiarui Fang authored Jul 06, 2022
  
  52736205
- [checkpoint] checkpoint for ColoTensor Model (#1196) · f38006ea
  Jiarui Fang authored Jul 06, 2022
  
  f38006ea