Sze-qq
|
7f5e4592eb
|
Update Experiment result about Colossal-AI with ZeRO (#479)
* [readme] add experimental visualisation regarding ColossalAI with ZeRO (#476)
* Hotfix/readme (#478)
* add experimental visualisation regarding ColossalAI with ZeRO
* adjust newly-added figure size
|
3 years ago |
Frank Lee
|
83a847d058
|
[test] added rerun on exception for testing (#475)
* [test] added rerun on exception function
* polish code
|
3 years ago |
ver217
|
d70f43dd7a
|
embedding remove attn mask (#474)
|
3 years ago |
HELSON
|
7544347145
|
[MOE] add unitest for MOE experts layout, gradient handler and kernel (#469)
|
3 years ago |
ver217
|
1559c0df41
|
fix attn mask shape of gpt (#472)
|
3 years ago |
ver217
|
3cb3fc275e
|
zero init ctx receives a dp process group (#471)
|
3 years ago |
ver217
|
7e30068a22
|
[doc] update rst (#470)
* update rst
* remove empty rst
|
3 years ago |
HELSON
|
aff9d354f7
|
[MOE] polish moe_env (#467)
|
3 years ago |
HELSON
|
bccbc15861
|
[MOE] changed parallelmode to dist process group (#460)
|
3 years ago |
Frank Lee
|
8f9617c313
|
[release] update version (#465)
|
3 years ago |
Frank Lee
|
2963565ff8
|
[test] fixed release workflow step (#464)
|
3 years ago |
Frank Lee
|
292590e0fa
|
[test] fixed release workflow condition (#463)
|
3 years ago |
Frank Lee
|
90bd97b9c0
|
[devops] fixed workflow bug (#462)
|
3 years ago |
ver217
|
304263c2ce
|
fix gpt attention mask (#461)
|
3 years ago |
ver217
|
fc8e6db005
|
[doc] Update docstring for ZeRO (#459)
* polish sharded model docstr
* polish sharded optim docstr
* polish zero docstr
* polish shard strategy docstr
|
3 years ago |
HELSON
|
84fd7c1d4d
|
add moe context, moe utilities and refactor gradient handler (#455)
|
3 years ago |
Frank Lee
|
af185b5519
|
[test] fixed amp convergence comparison test (#454)
|
3 years ago |
ver217
|
a241f61b34
|
[zero] Update initialize for ZeRO (#458)
* polish code
* shard strategy receive pg in shard() / gather()
* update zero engine
* polish code
|
3 years ago |
ver217
|
642846d6f9
|
update sharded optim and fix zero init ctx (#457)
|
3 years ago |
Jiarui Fang
|
e2e9f82588
|
Revert "[zero] update sharded optim and fix zero init ctx" (#456)
* Revert "polish code"
This reverts commit 8cf7ff08cf .
* Revert "rename variables"
This reverts commit e99af94ab8 .
* Revert "remove surplus imports"
This reverts commit 46add4a5c5 .
* Revert "update sharded optim and fix zero init ctx"
This reverts commit 57567ee768 .
|
3 years ago |
ver217
|
8cf7ff08cf
|
polish code
|
3 years ago |
ver217
|
e99af94ab8
|
rename variables
|
3 years ago |
ver217
|
46add4a5c5
|
remove surplus imports
|
3 years ago |
ver217
|
57567ee768
|
update sharded optim and fix zero init ctx
|
3 years ago |
Frank Lee
|
f27d801a13
|
[test] optimized zero data parallel test (#452)
|
3 years ago |
github-actions[bot]
|
cfcc8271f3
|
[Bot] Automated submodule synchronization (#451)
Co-authored-by: github-actions <github-actions@github.com>
|
3 years ago |
Frank Lee
|
ac4513c56e
|
[DevOps] remove unneeded dependency in build workflow (#449)
|
3 years ago |
Jiarui Fang
|
0fcfb1e00d
|
[test] make zero engine test really work (#447)
|
3 years ago |
Frank Lee
|
bb2790cf0b
|
optimize engine and trainer test (#448)
|
3 years ago |
Jiarui Fang
|
237d08e7ee
|
[zero] hybrid cpu adam (#445)
|
3 years ago |
Frank Lee
|
b72b8445c6
|
optimized context test time consumption (#446)
|
3 years ago |
Jiarui Fang
|
496cbb0760
|
[hotfix] fix initialize bug with zero (#442)
|
3 years ago |
Frank Lee
|
725a39f4bd
|
update github CI with the current workflow (#441)
|
3 years ago |
Frank Lee
|
5a1e33b97f
|
update contributing.md with the current workflow (#440)
|
3 years ago |
Jiarui Fang
|
17b8274f8a
|
[unitest] polish zero config in unittest (#438)
|
3 years ago |
Jiarui Fang
|
640a6cd304
|
[refactory] refactory the initialize method for new zero design (#431)
|
3 years ago |
Frank Lee
|
4f85b687cf
|
[misc] replace codebeat with codefactor on readme (#436)
|
3 years ago |
Frank Lee
|
bffd85bf34
|
added testing module (#435)
|
3 years ago |
HELSON
|
dbdc9a7783
|
added Multiply Jitter and capacity factor eval for MOE (#434)
|
3 years ago |
Frank Lee
|
b03b3ae99c
|
fixed mem monitor device (#433)
fixed mem monitor device
|
3 years ago |
Frank Lee
|
14a7094243
|
fixed fp16 optimizer none grad bug (#432)
|
3 years ago |
ver217
|
fce9432f08
|
sync before creating empty grad
|
3 years ago |
ver217
|
ea6905a898
|
free param.grad
|
3 years ago |
ver217
|
9506a8beb2
|
use double buffer to handle grad
|
3 years ago |
Frank Lee
|
0f5f5dd556
|
fixed gpt attention mask in pipeline (#430)
|
3 years ago |
Jiarui Fang
|
f9c762df85
|
[test] merge zero optim tests (#428)
|
3 years ago |
Frank Lee
|
f0d6e2208b
|
[polish] add license meta to setup.py (#427)
|
3 years ago |
Jiarui Fang
|
5d7dc3525b
|
[hotfix] run cpu adam unittest in pytest (#424)
|
3 years ago |
Jiarui Fang
|
54229cd33e
|
[log] better logging display with rich (#426)
* better logger using rich
* remove deepspeed in zero requirements
|
3 years ago |
HELSON
|
3f70a2b12f
|
removed noisy function during evaluation of MoE router (#419)
|
3 years ago |