Commit Graph

  • a2a79c7992 Add function display_and_save_batch in egs/wenetspeech/ASR/pruned_transducer_stateless2/train.py suxia 2022-08-10 17:59:31 +08:00
  • 56afb4aa3c Fix comments pkufool 2022-08-10 17:53:49 +08:00
  • 1138b27f16 modify pretrained.py yaozengwei 2022-08-10 17:16:09 +08:00
  • 88400a4435 Fix torch.jit.script pkufool 2022-08-10 16:49:35 +08:00
  • 8f3645e5cb modify test_model.py, test if the model can be successfully exported by jit.trace yaozengwei 2022-08-10 16:20:25 +08:00
  • 89124a59ac Convert make_pad_mask Fangjun Kuang 2022-08-10 16:05:31 +08:00
  • 522a45ce75 add using averaged model in export.py yaozengwei 2022-08-10 15:43:50 +08:00
  • f51e64dada add cut_id to results for all recipes pkufool 2022-08-10 14:52:12 +08:00
  • 317f47fb37 test RelPositionalEncoding Fangjun Kuang 2022-08-09 23:32:28 +08:00
  • 6217b5e64c Merge branch 'master' into sort_results pkufool 2022-08-09 22:23:18 +08:00
  • 10360bed41 Test encoder_embed Fangjun Kuang 2022-08-09 20:03:38 +08:00
  • 45c7894111 support jit trace yaozengwei 2022-08-09 19:29:21 +08:00
  • 5149788cb2
    Fix computing averaged loss in the aishell recipe. (#523) Fangjun Kuang 2022-08-09 10:53:31 +08:00
  • 365c6aa045 Support converting RelPositionalEncoding. Fangjun Kuang 2022-08-08 15:27:04 +08:00
  • 4ec243c730 begin to add documents for pruned_transducer_stateless pkufool 2022-08-08 10:41:13 +08:00
  • aa078cc6d7 Add cut_id to recognition results pkufool 2022-08-07 17:19:19 +08:00
  • dc6499a052 Merge branch 'master' into sort_results pkufool 2022-08-07 15:41:31 +08:00
  • 865fd211f7 Added export.py Guanbo Wang 2022-08-07 04:35:41 +00:00
  • b13b20b086 Added decoding scripts Guanbo Wang 2022-08-07 04:34:12 +00:00
  • f7c3d1ee28
    Merge 561c428e508da15074afb4b3496f5f6aafbe65d5 into f24b76e64bb59e157e3904d0330a132a308f18c0 Shayne Mei 2022-08-07 12:00:32 +08:00
  • b4aa7b0a7e Filter non-finite losses Fangjun Kuang 2022-08-07 06:46:53 +08:00
  • f24b76e64b
    fix torchaudio version (#524) FNLPprojects 2022-08-06 18:33:43 +08:00
  • 595aabce85 fix torchaudio version xinzhulin 2022-08-06 18:32:30 +08:00
  • 21dea66071 fix torchaudio version xinzhulin 2022-08-06 18:04:00 +08:00
  • 5c33899ddc Increase cov_min[3] from 0.001 to 0.002 Daniel Povey 2022-08-06 16:28:02 +08:00
  • f1a664e0d4 Set find_unused_parameters optionally. Fangjun Kuang 2022-08-06 12:52:25 +08:00
  • 540020bdc8 Fix computing averaged loss in the aishell recipe. Fangjun Kuang 2022-08-06 12:25:11 +08:00
  • 1f7832b93c
    Fix loading sampler state dict. (#421) Fangjun Kuang 2022-08-06 10:00:08 +08:00
  • 9bbf8ada57 Scale up diag of grad_cov by 1.0001 prior to diagonalizing it. Daniel Povey 2022-08-06 07:06:23 +08:00
  • c021b4fec6 Increase cov_min[3] from 0.0001 to 0.001 Daniel Povey 2022-08-06 07:02:52 +08:00
  • d69d83a83e test conformer.encoder_embed Fangjun Kuang 2022-08-05 21:28:05 +08:00
  • 0615fb316f add missing file Fangjun Kuang 2022-08-05 20:49:42 +08:00
  • 280e1c0312 Finishing testing encoder_embedding. Fangjun Kuang 2022-08-05 20:35:23 +08:00
  • 03b056ca37 add scaling converter for ScaledLSTM yaozengwei 2022-08-05 19:38:30 +08:00
  • 9e4b5bdfdf Merge remote-tracking branch 'k2-fsa/master' into lstm yaozengwei 2022-08-05 14:36:04 +08:00
  • 4ace4f2ed0 Sort result to make it more convenient to compare decoding results pkufool 2022-08-05 12:22:43 +08:00
  • a5b9b7b974 Cosmetic changes Daniel Povey 2022-08-05 03:49:54 +08:00
  • 7157f62af3
    Merging onnx models (#518) Yunusemre 2022-08-04 18:03:41 +03:00
  • 41b983ba7b fix minor Yunus Emre Özköse 2022-08-04 16:40:00 +03:00
  • b406c1beff Test ScaledConv2D. Fangjun Kuang 2022-08-04 21:39:20 +08:00
  • 4c3346bf6c revert to previous requirements.txt Yunus Emre Özköse 2022-08-04 15:00:14 +03:00
  • a4dd273776
    fix about tensorboard (#516) Zengwei Yao 2022-08-04 19:57:12 +08:00
  • a213600065 fix instaliing onnx_graphsurgeon Yunus Emre Özköse 2022-08-04 13:32:22 +03:00
  • f291599b4d fix installing onnx_graphsurgeon Yunus Emre Özköse 2022-08-04 12:04:25 +03:00
  • 19d5435afc fix input/output names Yunus Emre Özköse 2022-08-04 09:11:21 +03:00
  • 40126355ec fix requirements Yunus Emre Özköse 2022-08-04 08:36:00 +03:00
  • e538232485
    change for pruned rnnt5 train.py (#519) Mingshuang Luo 2022-08-04 12:29:39 +08:00
  • f061cac4f3 Change default ctc loss scale value from 0.5 to 0.3 Fangjun Kuang 2022-08-04 12:06:22 +08:00
  • e07adc70a9 change for pruned rnnt5 train.py luomingshuang 2022-08-04 11:51:19 +08:00
  • dc9133227f Reworked how inverse is done, fixed bug in _apply_min_max_with_metric, regarding how M is normalized. Daniel Povey 2022-08-04 09:46:14 +08:00
  • 18a6b8afad fix style Yunus Emre Özköse 2022-08-03 17:10:28 +03:00
  • d2de493cd8 fix style Yunus Emre Özköse 2022-08-03 17:05:16 +03:00
  • d382455945 add onnx-all-in-one test Yunus Emre Özköse 2022-08-03 16:50:55 +03:00
  • 6d3ad6e9dd remove unused arguments Yunus Emre Özköse 2022-08-03 16:50:21 +03:00
  • 390aeca769 minor fix Yunus Emre Özköse 2022-08-03 16:23:51 +03:00
  • f3a546e04f add onnx_check script for all-in-one onnx model Yunus Emre Özköse 2022-08-03 15:37:56 +03:00
  • 183821e6a0 add export function of onnx-all-in-one to export.py Yunus Emre Özköse 2022-08-03 15:37:23 +03:00
  • 36eacaccb2
    Fix preparing char based lang and add multiprocessing for wenetspeech text segmentation (#513) Weiji Zhuang 2022-08-03 19:19:40 +08:00
  • 71871d5614 fix style WeijiZhuang 2022-08-03 18:24:07 +08:00
  • 6af5a82d8f
    Convert ScaledEmbedding to nn.Embedding for inference. (#517) Fangjun Kuang 2022-08-03 15:34:55 +08:00
  • c9bf1f49db fix style yaozengwei 2022-08-03 15:30:08 +08:00
  • a7021400c6 Fix CI style issues. Fangjun Kuang 2022-08-03 15:26:05 +08:00
  • e130e41b83 Convert ScaledEmbedding to nn.Embedding for inference. Fangjun Kuang 2022-08-03 15:24:57 +08:00
  • 16ea305fbf fix metricstracker yaozengwei 2022-08-03 15:14:17 +08:00
  • 561c428e50
    Merge branch 'master' into use_explicit_relative_imports Shayne Mei 2022-08-02 22:02:58 -07:00
  • 58a96e5b68
    Support exporting to ONNX format (#501) Fangjun Kuang 2022-08-03 10:30:28 +08:00
  • ca19b98949 sort imports shaynemei 2022-08-02 17:45:23 -07:00
  • 5657fa44a4 use explicit relative imports shaynemei 2022-08-02 17:44:36 -07:00
  • 887819cd8a
    Merge 1f843e18f227639444dd8c6ae16d504684d2e452 into 132132f52a660f74dac552b432776d82aa1f5dab Fangjun Kuang 2022-08-02 18:35:58 -04:00
  • 766bf69a98 Reduce cov_max[2] from 4.0 to 3.5 Daniel Povey 2022-08-03 04:10:11 +08:00
  • fffd3bbd9e Fix preparing char based lang for wenetspeech WeijiZhuang 2022-08-02 23:40:27 +08:00
  • b986ef5ddc add multiprocessing for wenetspeech text segmentation WeijiZhuang 2022-08-02 23:38:41 +08:00
  • 5c2c91e53f fix Guanbo Wang 2022-08-02 15:38:39 +00:00
  • 132132f52a
    liear_fst_with_self_loops (#512) LIyong.Guo 2022-08-02 22:28:12 +08:00
  • 0fbaa0d32f liear_fst_with_self_loops Guo Liyong 2022-08-02 21:34:22 +08:00
  • 72dc1780d6 Fix as_strided for streaming conformer. Fangjun Kuang 2022-08-02 20:39:55 +08:00
  • 129b28aa9b Increase cov_min[2] from 0.05 to 0.1; decrease cov_max[2] from 5.0 to 4.0. Daniel Povey 2022-08-02 15:17:24 +08:00
  • 202752418a Increase cov_min[2] from 0.02 to 0.05. Daniel Povey 2022-08-02 15:15:41 +08:00
  • e44ab25e99 Bug fix Daniel Povey 2022-08-02 14:31:37 +08:00
  • 10846568b9 use explicit relative imports for librispeech shaynemei 2022-08-01 21:39:21 -07:00
  • 8714df792a use explicit relative imports for aimeeting shaynemei 2022-08-01 21:39:09 -07:00
  • 6df69603a8 use explicit relative imports for aishell4 shaynemei 2022-08-01 21:37:24 -07:00
  • 4cb62a3202 use explicit relative imports for aishell2 shaynemei 2022-08-01 21:35:06 -07:00
  • 51f2e377ba use explicit relative imports for aishell shaynemei 2022-08-01 21:34:58 -07:00
  • dd25072b3b use explicit relative imports for aidatatang_200zh shaynemei 2022-08-01 21:34:31 -07:00
  • 2d85fd103d use explicit relative imports shaynemei 2022-08-01 21:24:16 -07:00
  • e9f4ada1c0 Swap the order of applying min and max in smoothing operations Daniel Povey 2022-08-02 11:55:43 +08:00
  • 7daf148342 use explicit relative imports for aidatatang_200zh shaynemei 2022-08-01 20:54:59 -07:00
  • 60621b242a sort imports shaynemei 2022-08-01 20:49:52 -07:00
  • 9149a92dd8 use explicit relative imports shaynemei 2022-08-01 20:46:10 -07:00
  • e7b5a8ecbb sort imports shaynemei 2022-08-01 20:37:16 -07:00
  • 9473c7e23d Lots of changes to how min and max are applied, use 1-norm for min in smooth_cov but not _apply_min_max_with_metric. Daniel Povey 2022-08-02 11:29:54 +08:00
  • f2b72c80ff flake8 Guanbo Wang 2022-08-02 02:20:13 +00:00
  • 9701cf081c more links Guanbo Wang 2022-08-02 02:14:31 +00:00
  • 71dea95287 Added train.py Guanbo Wang 2022-08-02 02:04:04 +00:00
  • 6ab4cf615d 1st draft of new method of normalizing covs that uses normalization w.r.t. spectral 2-norm Daniel Povey 2022-08-02 09:34:37 +08:00
  • 10aaa1ffd1 link the files from librispeech/ASRpruned_transducer_stateless5 Guanbo Wang 2022-08-02 01:24:34 +00:00
  • 4919134a94 Merge making hidden_dim an arg Daniel Povey 2022-08-02 09:09:29 +08:00
  • c64bd5ebcd Merge making hidden_dim an arg Daniel Povey 2022-08-02 09:07:36 +08:00
  • b008340d83 Merge making hidden_dim an arg Daniel Povey 2022-08-02 09:01:19 +08:00