Commit Graph

42 Commits (913b2300c30b4816429abcd03e790d5a81d0610e)

Author SHA1 Message Date
Hui Zhang 913b2300c3 nprocs 0 for cpu, other for gpu
3 years ago
Hui Zhang 98b15eda05 batch WaveDataset
3 years ago
Hui Zhang bab29b94f1 fix train log
3 years ago
Hui Zhang daaa72a606 resuem train with epoch and iteration increase
3 years ago
Hui Zhang 7ae204eb0f fix log for u2
3 years ago
Hui Zhang 282914f45b
Merge branch 'develop' into fix_varbase
3 years ago
Hui Zhang 231499a347 VarBase.__getitem__ work for np.int64, np.longlong; but __setitem_varbase__ not support paddle.int16/set_value op not support
3 years ago
Hui Zhang 000183ea49 tensor.size to tensor.shape
3 years ago
Hui Zhang 054e099b28 format
3 years ago
Hui Zhang 0e91d26ae3 fix log; add report to trainer
3 years ago
Hui Zhang d7a33b9d5e update config with contextlib
3 years ago
Hui Zhang cda6ca8323 add benchmark flags, and logic
3 years ago
Hui Zhang 3a5258f6a0 lr and opt param will restore from ckpt, so we do not set lr manully
3 years ago
Hui Zhang 5fdda953b9 add op profiling
3 years ago
Hui Zhang 256e9c1b9c more doc for egs
3 years ago
Hui Zhang 3843372958 u2 with chianer updater
3 years ago
Hui Zhang 8873ebe38c add timer for u2; refactor grad norm type
3 years ago
Hui Zhang 7e136d0893 support no_sync for backward; ds support accum grad
3 years ago
Hui Zhang 561d5cf085 refactor feature, dict and argument for new config format
3 years ago
Hui Zhang 99dfe04515 test w/ all example
3 years ago
Hui Zhang e76123d418 rm useless
3 years ago
Hui Zhang 820b4db287 with all args for scheduler
3 years ago
Hui Zhang 3912c255ef support noam lr and opt
3 years ago
Hui Zhang 1cd4d4bf83 fix tiny conf and refactor optimizer and scheduler
3 years ago
Hui Zhang 6ee67785f6 fix ctc alignment
3 years ago
Hui Zhang 7ec623f7ae Merge branch 'develop' into align
3 years ago
Hui Zhang 9b3acddd5d fix conf for new datapipe; u2 export inputspec
3 years ago
Hui Zhang 90788b116d more comment; fix datapipe of align
3 years ago
Hui Zhang 1e2a5887aa Merge branch 'develop' into align
3 years ago
Haoxin Ma d55e6b5a0a revise from_pretrained function
3 years ago
Haoxin Ma 3a743f3717 fix pre-commit
3 years ago
Haoxin Ma 557427736e move redundant params
3 years ago
Haoxin Ma 698d7a9bdb move batch_size, work_nums, shuffle_method, sortagrad to collator
3 years ago
Haoxin Ma 89a00eabeb revise deepspeech/exps/u2/model.py
3 years ago
Haoxin Ma 6ee3033cc4 finish aishell/s0
3 years ago
Haoxin Ma 7bae32f384 revise example/ting/s1
3 years ago
Haoxin Ma b4bda290aa fix bugs
3 years ago
Haoxin Ma a58b1cb30a add result output
3 years ago
Hui Zhang 30aba26693 add align code
4 years ago
Hui Zhang 0a7958b3f1
add tarball utils (#626)
4 years ago
Hui Zhang 0a3a840bee
more decoding method (#618)
4 years ago
Hui Zhang 71e046b0ba
E2E/Streaming Transformer/Conformer ASR (#578)
4 years ago