19 Commits

Author	SHA1	Message	Date
Fangjun Kuang	902dc2364a	Update docker for torch 2.1 (#1326)	2023-10-22 23:25:06 +08:00
Fangjun Kuang	f14b673408	Add HLG decoding with OpenFst on CPU for aishell conformer_ctc (#1279)	2023-10-01 13:46:16 +08:00
Fangjun Kuang	2318c3fbd0	Support CTC decoding on CPU using OpenFst and kaldi decoders. (#1244)	2023-09-26 16:36:19 +08:00
Fangjun Kuang	d6b28a11a7	Add export script for the yesno recipe. (#1212)	2023-08-11 23:57:00 +08:00
Fangjun Kuang	6fd674312c	Fix failed CI tests (#1166)	2023-07-05 10:52:34 +08:00
Yifan Yang	7c4ff66a3d	Fix yesno Cl test (#1078)	2023-05-22 12:46:43 +08:00
Fangjun Kuang	3883e362ad	Fix yesno CI test (#1077)	2023-05-22 12:29:51 +08:00
marcoyang1998	34d1b07c3d	Modified beam search with RNNLM rescoring (#1002) ... * add RNNLM rescore * add shallow fusion and lm rescore for streaming zipformer * minor fix * update RESULTS.md * fix yesno workflow, change from ubuntu-18.04 to ubuntu-latest	2023-04-17 16:43:00 +08:00
Yifan Yang	28af269e5e	Fix for workflow (#934)	2023-03-09 17:38:15 +08:00
Fangjun Kuang	6533f359c9	Fix CI (#726) ... * Fix CI * Disable shuffle for yesno. See https://github.com/k2-fsa/icefall/issues/197	2022-12-02 10:53:06 +08:00
Mingshuang Luo	beab229fd7	[Ready to merge] Pruned_transducer_stateless2 for alimeeting dataset (#378) ... * add pruned-rnnt2 recipe for alimeeting dataset * update code for merging * change LilcomHdf5Writer to ChunkedLilcomHdf5Writer * change for test.yml * change for test.yml * change for test.yml * change for workflow yml * change for yml * change for yml * change for README.md * change for yml * solve the conflicts * solve the conflicts	2022-06-04 13:47:46 +08:00
Fangjun Kuang	a7643301ec	Cache pip packages for GitHub actions (#253) ... * Cache pip packages in GitHub actions.	2022-03-15 15:34:21 +08:00
Fangjun Kuang	336283f872	New label smoothing (#109) ... * Modify label smoothing to match the one implemented in PyTorch. * Enable CI for torch 1.10 * Fix CI errors. * Fix CI installation errors. * Fix CI installation errors. * Minor fixes. * Minor fixes. * Minor fixes. * Minor fixes. * Minor fixes. * Fix CI errors.	2021-11-17 19:24:07 +08:00
Fangjun Kuang	a80e58e15d	Refactor decode.py to make it more readable and more modular. (#44) ... * Refactor decode.py to make it more readable and more modular. * Fix an error. Nbest.fsa should always have token IDs as labels and word IDs as aux_labels. * Add nbest decoding. * Compute edit distance with k2. * Refactor nbest-oracle. * Add rescore with nbest lists. * Add whole-lattice rescoring. * Add rescoring with attention decoder. * Refactoring. * Fixes after refactoring. * Fix a typo. * Minor fixes. * Replace [] with () for shapes. * Use k2 v1.9 * Use Levenshtein graphs/alignment from k2 v1.9 * [doc] Require k2 >= v1.9 * Minor fixes.	2021-09-20 15:44:54 +08:00
Fangjun Kuang	cc77cb3459	Fix decode.py to remove the correct axis. (#50) ... * Fix decode.py to remove the correct axis. * Run GitHub actions manually.	2021-09-17 16:49:03 +08:00
Wei Kang	9a6e0489c8	update api for RaggedTensor (#45) ... * Fix code style * update k2 version in CI * fix compile hlg	2021-09-14 16:39:56 +08:00
Fangjun Kuang	abadc71415	Use new APIs with k2.RaggedTensor (#38) ... * Use new APIs with k2.RaggedTensor * Fix style issues. * Update the installation doc, saying it requires at least k2 v1.7 * Use k2 v1.7	2021-09-08 14:55:30 +08:00
Fangjun Kuang	57cb611665	[yesno] Remove padding in TDNN (#21) ... * Disable SpecAug for yesno. Also replace Adam with SGD. * Remove padding in the model to make the results reproducible.	2021-08-23 15:59:36 +08:00
Fangjun Kuang	6c2c9b9d74	Add recipe for the yes_no dataset. (#16) ... * Add recipe for the yes_no dataset. * Refactoring: Remove unused code. * Add Colab notebook for the yesno dataset. * Add GitHub actions to run yesno. * Fix a typo. * Minor fixes. * Train more epochs for GitHub actions. * Minor fixes. * Minor fixes. * Fix style issues.	2021-08-23 11:36:29 +08:00