- 11 Mar, 2022 40 commits
-
-
jiaruifang authored
-
jiaruifang authored
-
jiaruifang authored
-
jiaruifang authored
-
Frank Lee authored
-
Frank Lee authored
-
Jie Zhu authored
* fix merge conflict modify unit test remove unnessesary log info reformat file * remove unused module * remove unnecessary sync function * change doc string style from Google to Sphinx
-
ver217 authored
-
jiaruifang authored
-
jiaruifang authored
-
Jiarui Fang authored
-
Frank Lee authored
-
HELSON 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
-
binmakeswell authored
-
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
-
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
-
FrankLeeeee authored
-
FrankLeeeee authored
-
FrankLeeeee authored
-
ver217 authored
-
ver217 authored
-
ver217 authored
-
ver217 authored
-
Jiarui Fang authored
* init shard param from shape tuple * add more unitest for shard param
-
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
-
Frank Lee authored
* added unit test for sharded optimizer * refactor for elegance
-
Frank Lee authored
-
Jiarui Fang authored
-
Jie Zhu authored
* bug fix: pass hook_list to engine * change parameter name
-