Commit Graph

49 Commits (b3c5104499ddda4b7a05a63602ba003e1a5668fe)

Author SHA1 Message Date
Hui Zhang feaf71d468 u2 kaldi mutli process test with batchsize one
3 years ago
Hui Zhang dd06472432 extract ctc_align into ctc_utils
3 years ago
Hui Zhang 8b45c3e65e refactor trainer.py and rm ueseless dir setup code
3 years ago
Hui Zhang 466672e1de no_sync if paddle support else nullcontext
3 years ago
Hui Zhang e0f91b0994
Merge pull request #865 from PaddlePaddle/collator
3 years ago
Hui Zhang 856d641c9c multi worker for dataloader
3 years ago
Hui Zhang 32aaf403b5
Merge pull request #861 from PaddlePaddle/ctc
3 years ago
Hui Zhang f7d7e70cb2 more ctc check; valid dataloader with num workers
3 years ago
huangyuxin 9c37d10992 optimize the log
3 years ago
Hui Zhang 9d5eb74066 fix decode json file
3 years ago
Hui Zhang ae87bc8c7a dump decode result as jsonlines
3 years ago
Hui Zhang c6e8a33b73 fix set_device; more utils; args.opts support multi same name
3 years ago
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 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 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
4 years ago
Hui Zhang 7ec623f7ae Merge branch 'develop' into align
4 years ago
Hui Zhang 9b3acddd5d fix conf for new datapipe; u2 export inputspec
4 years ago
Hui Zhang 90788b116d more comment; fix datapipe of align
4 years ago
Hui Zhang 1e2a5887aa Merge branch 'develop' into align
4 years ago
Haoxin Ma d55e6b5a0a revise from_pretrained function
4 years ago
Haoxin Ma 3a743f3717 fix pre-commit
4 years ago
Haoxin Ma 557427736e move redundant params
4 years ago
Haoxin Ma 89a00eabeb revise deepspeech/exps/u2/model.py
4 years ago
Haoxin Ma 6ee3033cc4 finish aishell/s0
4 years ago
Haoxin Ma 7bae32f384 revise example/ting/s1
4 years ago
Haoxin Ma b4bda290aa fix bugs
4 years ago
Haoxin Ma a58b1cb30a add result output
4 years ago
Hui Zhang 30aba26693 add align code
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