Commit Graph

57 Commits (ed19e243dead2e78a343f59213d8017eccf598eb)

Author SHA1 Message Date
huangyuxin b1a90d4d7a add hub for s1 in aishell and librispeech
3 years ago
Hui Zhang 97136fe97f trainer eval when not train
3 years ago
Hui Zhang feaf71d468 u2 kaldi mutli process test with batchsize one
3 years ago
Hui Zhang d14e56476f multi workers for valid dataloader
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 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