Commit Graph

  • 86f45caafc
    Merge b52b5c683f4f7daef450eebf60f651100b916cdc into 5379c8e9fa13f6f2364b4a0db89fa3074266fb58 LIyong.Guo 2022-06-16 18:22:21 -04:00
  • cd14b3fe1c Update file names in Hubert VQ recipe Piotr Żelasko 2022-06-16 16:53:31 +00:00
  • f8541b3ab1 minor fix yaozengwei 2022-06-16 22:21:58 +08:00
  • 10ea04ae80 fix doc yaozengwei 2022-06-16 22:02:28 +08:00
  • 71a9c33bca RNN decoding refactoring Erwan 2022-06-16 11:17:51 +02:00
  • 496abc30c0 support using extracted codebook indexes yaozengwei 2022-06-16 21:31:26 +08:00
  • c1c893bd13 add pruned transducer stateless5 recipe for tal_csasr luomingshuang 2022-06-16 19:19:00 +08:00
  • 98785f63c0
    Merge 6a8ecc38685ecbbb966bc01cb4aa5e34c90ff243 into 5379c8e9fa13f6f2364b4a0db89fa3074266fb58 Fangjun Kuang 2022-06-16 17:31:23 +08:00
  • 5379c8e9fa
    Disable drop_last in testing time (#427) Wei Kang 2022-06-16 15:43:48 +08:00
  • 230ee5c991 Disable drop_last in testing time pkufool 2022-06-16 15:27:47 +08:00
  • 94e48ff07e Some code improvements... Daniel Povey 2022-06-16 10:43:26 +08:00
  • 00b1c291a6 Refactoring LM loader Erwan 2022-06-15 15:17:19 +02:00
  • 25d540a758 Remove DDP Erwan 2022-06-15 08:57:08 +02:00
  • 1ed96824a0 support to join manifest files yaozengwei 2022-06-15 19:42:41 +08:00
  • 2a47db653b Merge remote-tracking branch 'k2-fsa/master' into rand_combine yaozengwei 2022-06-15 14:43:52 +08:00
  • f9bb745cf3 Use exp_avg_sq as ref_exp_avg_sq Daniel Povey 2022-06-15 14:03:26 +08:00
  • 860322bf30 Make it less verbose; fix scale_speed setting; testing min_diag_smooth=1.0 for debug Daniel Povey 2022-06-15 13:02:04 +08:00
  • bc5d72b0f3 Fix potential bug re scale change Daniel Povey 2022-06-15 12:49:47 +08:00
  • 57957cc049 Add diagnostics Daniel Povey 2022-06-15 12:39:16 +08:00
  • 7d5eacf275 save only vq-related info to manifest yaozengwei 2022-06-15 12:19:56 +08:00
  • feb9fd05e8 Small fixes. Fangjun Kuang 2022-06-15 12:03:02 +08:00
  • 46430a5024 More fixes. Fangjun Kuang 2022-06-15 11:43:54 +08:00
  • d3af7eebbf Fix CI errors. Fangjun Kuang 2022-06-15 11:23:13 +08:00
  • df6ab85fb4 Fix CI errors. Fangjun Kuang 2022-06-15 10:56:06 +08:00
  • 0679b363b0 Some debug stuff, modify cond_eps for param_cov Daniel Povey 2022-06-14 22:38:51 +08:00
  • 5c3ee8bfcd
    [Ready to merge] Pruned transducer stateless5 recipe for AISHELL4 (#399) Mingshuang Luo 2022-06-14 22:19:05 +08:00
  • 932cedce59 Bug fix RE eps*eps; add/tune diagnostics Daniel Povey 2022-06-14 21:43:27 +08:00
  • 2615e48779 Some cleanup and configuration changes. Daniel Povey 2022-06-14 21:17:50 +08:00
  • c6f62dfc08 Version that is passing the tests Daniel Povey 2022-06-14 20:45:57 +08:00
  • fd7646d294 Fix loading sampler state dict. Fangjun Kuang 2022-06-14 17:40:18 +08:00
  • 1bf2e17437 Add fast_beam_search_nbest. Fangjun Kuang 2022-06-14 17:04:04 +08:00
  • 127066f33a Do a change for READMD.md luomingshuang 2022-06-14 15:24:06 +08:00
  • 779589a2de Add RNN params to parser Erwan 2022-06-14 08:49:50 +02:00
  • e597b68160 Improve roundoff properties of algorithm for getting C, use randomized testing. Daniel Povey 2022-06-14 12:59:41 +08:00
  • 1c4987f6e6 update codes for merging luomingshuang 2022-06-14 11:44:03 +08:00
  • d729e970e1 Some changes for stability Daniel Povey 2022-06-14 10:25:17 +08:00
  • 83fc25dc1c
    Merge b9d61199325eb699c6141e579b4a0f20582d55bc into 53f38c01d2f56e5608ed67212f3f28701ca3e27b LIyong.Guo 2022-06-13 16:30:35 +02:00
  • 193b44ed7a use average value as memory vector for each chunk yaozengwei 2022-06-13 22:14:24 +08:00
  • 1c067e7364 init files yaozengwei 2022-06-13 22:10:50 +08:00
  • ec8646d0cd Merge remote-tracking branch 'k2-fsa/master' yaozengwei 2022-06-13 20:55:28 +08:00
  • 413ca2ef77 Merge remote-tracking branch 'origin/master' into streaming-conformer pkufool 2022-06-13 19:10:44 +08:00
  • 6a8ecc3868 Add pruned RNN-T multi-dataset setup for aishell Fangjun Kuang 2022-06-13 19:05:03 +08:00
  • 09514259b8 Add pruned RNN-T multidateset setup for aishell. Fangjun Kuang 2022-06-13 18:55:45 +08:00
  • d1e96afce2 Improve speed by messing with configuration, removing asserts. Daniel Povey 2022-06-13 17:27:34 +08:00
  • a41c4b6c9b Bug fixes Daniel Povey 2022-06-13 17:20:46 +08:00
  • 41b0e582df Bug fix to typo in optim.py Daniel Povey 2022-06-13 16:10:49 +08:00
  • ca7cffcb42 Remove Decorrelate() class Daniel Povey 2022-06-13 16:08:32 +08:00
  • 7338c60296 Remove Decorrelate() Daniel Povey 2022-06-13 16:07:15 +08:00
  • c1f487e36d Move optim2.py to optim.py; use this optimizer in train.py Daniel Povey 2022-06-13 16:05:46 +08:00
  • 4e0d8c45f4 Move optim2.py to optim.py; use this optimizer in train.py Daniel Povey 2022-06-13 16:05:22 +08:00
  • 53f38c01d2
    Emformer with conv module and scaling mechanism (#389) Zengwei Yao 2022-06-13 15:09:17 +08:00
  • d5b90f278e Adding draft of new optimizer Daniel Povey 2022-06-13 15:01:10 +08:00
  • de1fe9dac4 update README.md yaozengwei 2022-06-13 13:42:12 +08:00
  • 2a5a70e03e Merge remote-tracking branch 'k2-fsa/master' yaozengwei 2022-06-13 12:52:28 +08:00
  • a1cbe1fd9c fix doc of stack and unstack, test case with batch_size=1 yaozengwei 2022-06-13 12:47:35 +08:00
  • adcbb4076d minor change of test_emformer.py yaozengwei 2022-06-12 18:11:36 +08:00
  • e00ad8104e add copyright info yaozengwei 2022-06-12 18:07:09 +08:00
  • 4dc0af1144 add test function for conv module, and minor fix. yaozengwei 2022-06-12 18:01:21 +08:00
  • b24453434f fix doc yaozengwei 2022-06-12 11:53:59 +08:00
  • 6b0a519976 fix typo yaozengwei 2022-06-12 11:39:42 +08:00
  • 41fae5855f add info yaozengwei 2022-06-12 11:37:52 +08:00
  • ea0a4ed627 fix typo yaozengwei 2022-06-12 11:23:32 +08:00
  • 1f5216236f update RESULTS.md yaozengwei 2022-06-12 11:21:43 +08:00
  • 7f94b86bb0 minor fix doc yaozengwei 2022-06-11 22:44:06 +08:00
  • 64c9053e39 minor fix yaozengwei 2022-06-11 22:40:40 +08:00
  • e011812bab minor fix doc yaozengwei 2022-06-11 22:31:55 +08:00
  • f2974e0df5 fix style yaozengwei 2022-06-11 22:24:12 +08:00
  • 86b9210d7b modify doc yaozengwei 2022-06-11 22:22:46 +08:00
  • 8c03d6079a minor fix yaozengwei 2022-06-11 22:12:37 +08:00
  • ce008aa2ca delete other test functions yaozengwei 2022-06-11 22:09:53 +08:00
  • 448a94c00a add author info yaozengwei 2022-06-11 22:09:16 +08:00
  • 626a2912ca modify export.py yaozengwei 2022-06-11 22:03:19 +08:00
  • 1a9af61497 copy from pruned_transducer_stateless4 yaozengwei 2022-06-11 21:40:19 +08:00
  • f57f1b8a44 use torch.div yaozengwei 2022-06-11 21:21:49 +08:00
  • c8adbcce64 support torch.jit.export yaozengwei 2022-06-11 21:04:30 +08:00
  • aaabe9f611 Merge branch 'emformer_conv_scale_new' into emformer_conv_scale yaozengwei 2022-06-11 18:34:14 +08:00
  • eadd0f6aa7 Merge remote-tracking branch 'k2-fsa/master' into emformer_conv_scale_new yaozengwei 2022-06-11 18:32:49 +08:00
  • 7e7bc22cb5 Add RNNLM for rescoring. Fangjun Kuang 2021-11-24 14:52:49 +08:00
  • a3732a2bf3 Start training LM for LibriSpeech. Fangjun Kuang 2021-11-23 11:58:25 +08:00
  • b41ad93816 Generate LM training data for the LibriSpeech recipe. Fangjun Kuang 2021-11-22 21:04:11 +08:00
  • a9bb9d4eca Add scripts for training and perplexity computation. Fangjun Kuang 2021-11-22 20:05:18 +08:00
  • eb56424c9b WIP: Begin to add RNNLM. Fangjun Kuang 2021-11-21 15:58:05 +08:00
  • 8c4b5454e7 Remove unnecessary line Daniel Povey 2022-06-11 13:14:18 +08:00
  • 73fd2fcd77 Updates to cain to new param_rms per element Daniel Povey 2022-06-11 13:08:08 +08:00
  • d301f8ac6c Merge Decorrelate work, and simplification to RandomCombine, into pruned_transducer_stateless7 Daniel Povey 2022-06-11 11:07:07 +08:00
  • 30379bb35d Merge branch 'cain3d_merge_supersimple' into merge_decorr_cain Daniel Povey 2022-06-11 10:51:30 +08:00
  • a3c97554e8 Reduce scale from 0.1 to 0.05 Daniel Povey 2022-06-10 23:34:46 +08:00
  • 2bbc63a2f5 Change first1k to decay1k Daniel Povey 2022-06-10 23:33:16 +08:00
  • f233b16974 add modify_beam_search, fast_beam_search yaozengwei 2022-06-10 22:10:43 +08:00
  • 41cdbae753 Increase scale from 0.05 to 0.1 Daniel Povey 2022-06-10 19:07:48 +08:00
  • 85ddfd958c Minor fixes pkufool 2022-06-10 19:06:06 +08:00
  • 5fb64a59b8 Change beta from 0.8 to 0.95 Daniel Povey 2022-06-10 19:05:04 +08:00
  • 6ed181595b Scale by grad norm Daniel Povey 2022-06-10 18:34:42 +08:00
  • 6a47bf1178 Decrease scale to 0.05 Daniel Povey 2022-06-10 18:33:50 +08:00
  • 026fb22076 fix bugs, no grad, and num_proccessed_frames yaozengwei 2022-06-10 17:17:03 +08:00
  • 9d4633facf Simplified gradient scaling [no scaling]; only use 1k first iters; beta =0.8 Daniel Povey 2022-06-10 17:16:05 +08:00
  • cecd52155c Apply x row scaling with grad Daniel Povey 2022-06-10 16:42:52 +08:00
  • 86c2d0fcc0 Make it run only for first 3k steps; larger scale; remove limit of 1.0 Daniel Povey 2022-06-10 16:33:22 +08:00
  • eeb95ed502 Fix issue with cov scale Daniel Povey 2022-06-10 16:25:45 +08:00
  • c671e213fc Increase beta from 0.95 to 0.98 Daniel Povey 2022-06-10 14:39:58 +08:00