- 14 Mar, 2022 2 commits
- 11 Mar, 2022 38 commits
-
-
Frank Lee authored
-
Jiarui Fang authored
* place params on cpu after zero init context * polish code * bucketzed cpu gpu tensor transter * find a bug in sharded optim unittest * add offload unittest for ShardedOptimV2. * polish code and make it more robust
-
Frank Lee authored
* refactored test with component func * fixed bug
-
LuGY authored
* Added activation offload * Fixed the import bug, used the pytest
-
Jiarui Fang authored
-
Jiarui Fang authored
-
jiaruifang authored
This reverts commit bef05489b642385c80e59fe757d598efd1752ecf.
-
Jiarui Fang authored
-
Jiarui Fang authored
* place params on cpu after zero init context * polish code
-
Jiarui Fang authored
-
Jiarui Fang authored
-
ver217 authored
-
HELSON authored
-
ver217 authored
-
ver217 authored
-
ver217 authored
-
jiaruifang authored
-
jiaruifang authored
-
jiaruifang authored
-
jiaruifang authored
-
ver217 authored
-
jiaruifang authored
-
jiaruifang authored
-
Jiarui Fang authored
-
Frank Lee authored
-
Jiarui Fang authored
* add zero init context * add more flags for zero init context fix bug of repeated converting param to ShardedParamV2 * polish code
-
1SAA authored
Fixed bug for learning rate scheduler
-
LuGY authored
* Added CPU Adam * finished the cpu adam * updated the license * delete useless parameters, removed resnet * modified the method off cpu adam unittest * deleted some useless codes * removed useless codes Co-authored-by:
ver217 <lhx0217@gmail.com> Co-authored-by:
Frank Lee <somerlee.9@gmail.com> Co-authored-by:
jiaruifang <fangjiarui123@gmail.com>
-
Jiarui Fang authored
-
Jiarui Fang authored
* init shard param from shape tuple * add more unitest for shard param * add set_payload method for ShardedParam * [zero] add shareded tensor class * polish code * add shard stratgy * move shard and gather logic to shard strategy from shard tensor. * polish code
-
ver217 authored
-
ver217 authored
-
Jiarui Fang authored
-
Jiarui Fang authored
* init shard param from shape tuple * add more unitest for shard param * add set_payload method for ShardedParam * [zero] add shareded tensor class * polish code
-
Jie Zhu authored
-
Jiarui Fang authored
* init shard param from shape tuple * add more unitest for shard param * add more unittests to shareded param
-
ver217 authored
-
Frank Lee authored
* added unit test for sharded optimizer * refactor for elegance
-