Commit Graph

  • f5f6903581 reformat Ryuk 2024-08-25 16:53:09 +08:00
  • 9865cd379f support transformer-lm when using lm rescoring Ryuk 2024-08-25 16:25:09 +08:00
  • 6eedf4d156 support transformer-lm in lm rescoring Ryuk 2024-08-25 16:25:09 +08:00
  • 50471d6f11
    Merge branch 'k2-fsa:master' into master Yu Lianjie 2024-08-24 12:20:54 +08:00
  • a6c02a4d8c
    zipformer BF16 training recipe (#1700) Xiaoyu Yang 2024-08-23 09:42:22 +08:00
  • 3b434fe83c
    fix triton onnx export (#1730) Yuekai Zhang 2024-08-23 09:33:46 +08:00
  • ad61d72dfa
    Update pretrain.py Yifan Yang 2024-08-22 18:32:33 +08:00
  • 6357d420b7
    Merge branch 'k2-fsa:master' into dev/k2ssl Yifan Yang 2024-08-22 18:29:05 +08:00
  • bb0f9735e7 fix triton onnx export Yuekai Zhang 2024-08-22 17:49:53 +08:00
  • 3fc06cc2b9
    Support AudioSet training with weighted sampler (#1727) Xiaoyu Yang 2024-08-22 15:27:25 +08:00
  • e94d4809fe minor changes marcoyang 2024-08-22 14:51:10 +08:00
  • 298230177a fix format marcoyang 2024-08-22 14:22:56 +08:00
  • f672df2a63
    Update run_multi_node_multi_gpu.sh Yifan Yang 2024-08-22 10:57:26 +08:00
  • 6dbcdbac5d
    Update pretrain.py Yifan Yang 2024-08-22 10:55:23 +08:00
  • 01a85c5335 update RESULTS.md marcoyang 2024-08-21 16:57:46 +08:00
  • fcf06872a2 support weighted sampler marcoyang 2024-08-21 16:47:42 +08:00
  • 1e34f0e2e0 add a script to compute sampling weight marcoyang 2024-08-21 16:37:21 +08:00
  • b585e14de3 minor changes marcoyang 2024-08-21 16:35:32 +08:00
  • dc353dcc7b remove full bf16 training, performance not good marcoyang 2024-08-21 15:31:44 +08:00
  • 8fe6713d10
    Update pretrain.py Yifan Yang 2024-08-21 12:53:19 +08:00
  • eca8afcb83
    Update pretrain.py Yifan Yang 2024-08-21 12:47:45 +08:00
  • d0a96a601c
    Update run_multi_node_multi_gpu.sh Yifan Yang 2024-08-21 12:42:06 +08:00
  • d025ce11ff use lr hours in librilight ssl yifanyeung 2024-08-19 23:14:26 +08:00
  • 70a1713662
    Merge branch 'k2-fsa:master' into dev/k2ssl Yifan Yang 2024-08-19 23:09:03 +08:00
  • 5952972294
    Keep the custom fields in libriheavy manifest (#1719) Xiaoyu Yang 2024-08-17 13:24:38 +08:00
  • 8b1402aab5
    Merge branch 'k2-fsa:master' into dev/k2ssl Yifan Yang 2024-08-16 20:19:58 +08:00
  • 6ac3343ce5
    fix path in README.md (#1722) Yifan Yang 2024-08-16 20:13:02 +08:00
  • 3d1f54c661
    fix path in README.md Yifan Yang 2024-08-16 20:05:30 +08:00
  • 60bfbffda7 Add pseudo-labeling based semi-supervised training recipe Han Zhu 2024-08-16 15:43:24 +08:00
  • 67a8896c79 fix format marcoyang 2024-08-15 11:47:27 +08:00
  • 01ab64685e add an extra flag for keeping the custom fields in the cuts required for PromptASR marcoyang 2024-08-15 11:38:54 +08:00
  • cce86a3943
    Merge branch 'k2-fsa:master' into dev/k2ssl Yifan Yang 2024-08-14 15:20:26 +08:00
  • 9b13eac946 fix misc line root 2024-08-14 15:47:28 +09:00
  • 814d3ac702 fix formatting issues root 2024-08-14 15:29:29 +09:00
  • 563292599b resolve PR issue root 2024-08-01 18:27:34 +09:00
  • 8d0107a959 remove prints root 2024-08-01 16:17:59 +09:00
  • 887366370d update for streaming root 2024-08-01 16:13:59 +09:00
  • cfef53dcbe remove streaming/greedy_search results folder root 2024-08-01 16:00:27 +09:00
  • eebe6add4c update streaming decode root 2024-08-01 15:44:50 +09:00
  • ed5fa02c5c update streaming decoding file root 2024-08-01 15:43:44 +09:00
  • d24f1e1bae update README for streaming root 2024-08-01 15:31:20 +09:00
  • 250ff30875 add streaming support to reazonresearch root 2024-08-01 15:08:31 +09:00
  • 1730fce688
    split save_results() -> save_asr_output() + save_wer_results() (#1712) Karel Vesely 2024-08-13 17:02:14 +02:00
  • c6b71417b8 fix open-commands path Yu Lianjie 2024-08-12 23:44:00 +08:00
  • 7d3c521818 split save_results() -> save_asr_output() + save_wer_results() Karel Vesely 2024-08-08 11:24:35 +02:00
  • d400bc5edf fix the CTC zipformer2 training Karel Vesely 2024-08-12 10:45:57 +02:00
  • a288d4123b minor changes marcoyang 2024-08-12 11:30:21 +08:00
  • f26dd3ba17 support multinode multigpu Yifan Yeung 2024-08-10 22:07:58 +08:00
  • 8e296b7047 add librilight ssl recipe Your Name 2024-08-09 18:54:10 +00:00
  • af67140ad2 minor changes marcoyang 2024-08-08 10:54:44 +08:00
  • a7854dddba support AMP training with bf16 marcoyang 2024-08-08 10:13:30 +08:00
  • d1974befaa remove decode_bf16 because it's not necessary: we can just use decode.py marcoyang 2024-08-08 10:10:22 +08:00
  • 916e84d726 resolve PR issue root 2024-08-01 18:27:34 +09:00
  • 8189d115bd remove prints root 2024-08-01 16:17:59 +09:00
  • 529d92ff4b Add docker images for torch 2.4 (#1704) Fangjun Kuang 2024-07-25 16:46:24 +08:00
  • db38ab044b Zipformer with Adam optimizer Han Zhu 2024-08-01 14:51:00 +08:00
  • e052481344 update for streaming root 2024-08-01 16:13:59 +09:00
  • 2d2daf66c2 remove streaming/greedy_search results folder root 2024-08-01 16:00:27 +09:00
  • 6317405ea0 update streaming decode root 2024-08-01 15:44:50 +09:00
  • 62eb090220 update streaming decoding file root 2024-08-01 15:43:44 +09:00
  • 5a0c247efa update README for streaming root 2024-08-01 15:31:20 +09:00
  • 5062f12c4f add streaming support to reazonresearch root 2024-08-01 15:08:31 +09:00
  • 3b257dd5ae
    Add docker images for torch 2.4 (#1704) Fangjun Kuang 2024-07-25 16:46:24 +08:00
  • ba7f743055 fix torchaudio Fangjun Kuang 2024-07-25 15:48:03 +08:00
  • 0de839d50b docker for torch 2.4 Fangjun Kuang 2024-07-25 15:29:26 +08:00
  • 7692bd4dfe add decoding script marcoyang 2024-07-24 15:06:10 +08:00
  • eb9800e121 minor updates marcoyang 2024-07-24 15:05:13 +08:00
  • f4912b283d remove the previous version marcoyang 2024-07-24 15:02:50 +08:00
  • 5a05da8fcc make modifications to support full bf16 training marcoyang 2024-07-23 23:14:32 +08:00
  • 0c29c45c32 add files marcoyang 2024-07-23 23:13:32 +08:00
  • 4af81af5a6
    Update Zipformer-xl 700M Results on multi-hans-zh (#1694) Yuekai Zhang 2024-07-18 21:05:59 +08:00
  • 936bc107de fix typo Yuekai Zhang 2024-07-18 15:26:06 +08:00
  • e9037721f4 update zipformer-xl results Yuekai Zhang 2024-07-18 15:17:49 +08:00
  • 11151415f3
    fix error in accum_grad (#1693) zzasdf 2024-07-17 17:47:43 +08:00
  • 3225978792 fix error in accum_grad zzasdf 2024-07-17 17:07:30 +08:00
  • 2e13298717
    Refactor ctc greedy search. (#1691) Fangjun Kuang 2024-07-15 12:01:47 +08:00
  • 6f22633a08 Refactor ctc greedy search. Fangjun Kuang 2024-07-15 10:14:15 +08:00
  • d47c078286
    add decoding method of ctc-greedy-search in zipformer recipe (#1690) Zengwei Yao 2024-07-14 17:30:13 +08:00
  • 77a1e76a5c add decoding method of ctc-greedy-search in zipformer recipe yaozengwei 2024-07-14 17:23:13 +08:00
  • 359ffce6c9
    Merge branch 'k2-fsa:master' into dev/zipformer_lstm Yifan Yang 2024-07-14 00:32:21 +08:00
  • 334beed2af
    fix usages of returned losses after adding attention-decoder in zipformer (#1689) Zengwei Yao 2024-07-12 16:50:58 +08:00
  • c58cf28358 fix usages of returned losses after adding attention-decoder in zipformer yaozengwei 2024-07-12 16:40:44 +08:00
  • f6febd658e
    "-" replace "_" fix writing error (#1687) Ziwei Li 2024-07-12 14:42:00 +08:00
  • 0defb8f120 "-" replace "_" fix writing error ziwei 2024-07-12 14:18:06 +08:00
  • 7225965a4a add blank penalty root 2024-07-12 00:01:04 +00:00
  • 19048e155b
    Cast grad_scale in whiten to float (#1663) Teo Wen Shen 2024-07-11 16:12:30 +09:00
  • d65187ec52
    Small fix (#1686) Yifan Yang 2024-07-11 14:45:35 +08:00
  • 9dc87d47d9 Small fix yfyeung 2024-07-11 13:51:12 +08:00
  • 61e60d90a6
    Merge branch 'k2-fsa:master' into dev/zipformer_lstm Yifan Yang 2024-07-10 15:21:23 +08:00
  • b796fdb883 update yfyeung 2024-07-10 00:21:03 -07:00
  • 7961b6bf23 update yfyeung 2024-07-10 00:16:24 -07:00
  • 785f3f0bcf
    Update RESULTS.md, adding results and model links of zipformer-small/medium CTC/AED models (#1683) Zengwei Yao 2024-07-09 20:04:47 +08:00
  • 42475269cb Update RESULTS.md, adding results and model links of zipformer-small/medium CTC/AED models yaozengwei 2024-07-09 20:02:09 +08:00
  • 1c3d992a39
    Update results using Zipformer-large on multi-hans-zh (#1679) Yuekai Zhang 2024-07-09 09:57:52 +08:00
  • f96bbdef41 update decode.py yfyeung 2024-07-06 08:13:20 -07:00
  • 2d64228efa
    Update attention_decoder.py (#1681) zr_jin 2024-07-06 09:01:34 +08:00
  • 325a825841
    Update requirements-ci.txt (#1682) zr_jin 2024-07-06 09:01:19 +08:00
  • 418075c782 Update requirements-ci.txt jinzr 2024-07-06 02:20:01 +08:00
  • 3cad98d69d Update attention_decoder.py jinzr 2024-07-06 02:14:21 +08:00
  • f76afff741
    Support CTC/AED option for Zipformer recipe (#1389) Zengwei Yao 2024-07-05 20:19:18 +08:00