Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Compare against torchaudio rnnt #1

Draft
wants to merge 2 commits into
base: master
Choose a base branch
from
Draft

Conversation

vincentqb
Copy link
Owner

This PR imports the RNN-T prototype in place of warp-transudcer or warp-rnnt and show that the difference is zero, and all tests are passing.

@vincentqb
Copy link
Owner Author

vincentqb commented Dec 24, 2020

Command with 7ba521b39508db26ceb82d122a7143e1b544e9b4

pytest test/test_e2e_asr_transducer.py -s

Output

82 passed, 2 warnings in 33.51s
============================= test session starts ==============================
platform linux -- Python 3.7.3, pytest-6.2.1, py-1.9.0, pluggy-0.13.1 -- /private/home/vincentqb/anaconda3/bin/python
cachedir: .pytest_cache
rootdir: /private/home/vincentqb/espnet, configfile: setup.cfg
plugins: typeguard-2.10.0, cov-2.10.1
collecting ... collected 82 items

test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic0-recog_dic0] --> diff 0.0 loss1 10.000711441040039 loss2 10.000711441040039
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic1-recog_dic1] --> diff 0.0 loss1 8.787003517150879 loss2 8.787003517150879
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic2-recog_dic2] --> diff 0.0 loss1 8.774653434753418 loss2 8.774653434753418
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic3-recog_dic3] --> diff 0.0 loss1 9.409162521362305 loss2 9.409162521362305
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic4-recog_dic4] --> diff 0.0 loss1 9.197712898254395 loss2 9.197712898254395
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic5-recog_dic5] --> diff 0.0 loss1 13.503780364990234 loss2 13.503780364990234
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic6-recog_dic6] --> diff 0.0 loss1 14.727216720581055 loss2 14.727216720581055
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic7-recog_dic7] --> diff 0.0 loss1 14.2301607131958 loss2 14.2301607131958
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic8-recog_dic8] --> diff 0.0 loss1 14.467546463012695 loss2 14.467546463012695
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic9-recog_dic9] --> diff 0.0 loss1 9.835400581359863 loss2 9.835400581359863
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic10-recog_dic10] --> diff 0.0 loss1 9.424392700195312 loss2 9.424392700195312
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic11-recog_dic11] --> diff 0.0 loss1 9.272937774658203 loss2 9.272937774658203
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic12-recog_dic12] --> diff 0.0 loss1 8.475971221923828 loss2 8.475971221923828
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic13-recog_dic13] --> diff 0.0 loss1 8.88613224029541 loss2 8.88613224029541
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic14-recog_dic14] --> diff 0.0 loss1 9.957626342773438 loss2 9.957626342773438
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic15-recog_dic15] --> diff 0.0 loss1 9.184822082519531 loss2 9.184822082519531
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic16-recog_dic16] --> diff 0.0 loss1 9.106863021850586 loss2 9.106863021850586
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic17-recog_dic17] --> diff 0.0 loss1 10.385744094848633 loss2 10.385744094848633
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic18-recog_dic18] --> diff 0.0 loss1 9.025452613830566 loss2 9.025452613830566
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic19-recog_dic19] --> diff 0.0 loss1 9.572765350341797 loss2 9.572765350341797
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic20-recog_dic20] --> diff 0.0 loss1 9.087263107299805 loss2 9.087263107299805
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic21-recog_dic21] --> diff 0.0 loss1 8.988186836242676 loss2 8.988186836242676
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic22-recog_dic22] --> diff 0.0 loss1 9.46450424194336 loss2 9.46450424194336
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic23-recog_dic23] --> diff 0.0 loss1 9.008488655090332 loss2 9.008488655090332
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic24-recog_dic24] --> diff 0.0 loss1 8.969382286071777 loss2 8.969382286071777
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic25-recog_dic25] --> diff 0.0 loss1 9.258123397827148 loss2 9.258123397827148
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic26-recog_dic26] --> diff 0.0 loss1 8.676462173461914 loss2 8.676462173461914
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic27-recog_dic27] --> diff 0.0 loss1 9.65167236328125 loss2 9.65167236328125
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic28-recog_dic28] --> diff 0.0 loss1 9.470110893249512 loss2 9.470110893249512
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic29-recog_dic29] --> diff 0.0 loss1 9.372766494750977 loss2 9.372766494750977
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic30-recog_dic30] --> diff 0.0 loss1 9.13366985321045 loss2 9.13366985321045
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic31-recog_dic31] --> diff 0.0 loss1 9.001062393188477 loss2 9.001062393188477
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic32-recog_dic32] --> diff 0.0 loss1 9.068403244018555 loss2 9.068403244018555
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic33-recog_dic33] --> diff 0.0 loss1 9.22912311553955 loss2 9.22912311553955
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic34-recog_dic34] --> diff 0.0 loss1 9.191285133361816 loss2 9.191285133361816
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic35-recog_dic35] --> diff 0.0 loss1 9.607641220092773 loss2 9.607641220092773
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic36-recog_dic36] --> diff 0.0 loss1 9.716028213500977 loss2 9.716028213500977
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic37-recog_dic37] --> diff 0.0 loss1 9.22525405883789 loss2 9.22525405883789
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic38-recog_dic38] --> diff 0.0 loss1 9.110225677490234 loss2 9.110225677490234
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic39-recog_dic39] --> diff 0.0 loss1 8.757622718811035 loss2 8.757622718811035
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic40-recog_dic40] --> diff 0.0 loss1 9.107383728027344 loss2 9.107383728027344
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic41-recog_dic41] --> diff 0.0 loss1 9.400981903076172 loss2 9.400981903076172
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic42-recog_dic42] --> diff 0.0 loss1 9.212616920471191 loss2 9.212616920471191
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic43-recog_dic43] --> diff 0.0 loss1 9.116785049438477 loss2 9.116785049438477
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic44-recog_dic44] --> diff 0.0 loss1 9.177767753601074 loss2 9.177767753601074
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic45-recog_dic45] --> diff 0.0 loss1 9.31392765045166 loss2 9.31392765045166
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic46-recog_dic46] --> diff 0.0 loss1 9.198881149291992 loss2 9.198881149291992
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic47-recog_dic47] --> diff 0.0 loss1 9.218058586120605 loss2 9.218058586120605
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic48-recog_dic48] --> diff 0.0 loss1 9.269806861877441 loss2 9.269806861877441
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic49-recog_dic49] --> diff 0.0 loss1 8.994089126586914 loss2 8.994089126586914
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic50-recog_dic50] --> diff 0.0 loss1 10.060941696166992 loss2 10.060941696166992
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic51-recog_dic51] --> diff 0.0 loss1 8.190975189208984 loss2 8.190975189208984
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic52-recog_dic52] --> diff 0.0 loss1 8.963150024414062 loss2 8.963150024414062
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic53-recog_dic53] --> diff 0.0 loss1 8.90825366973877 loss2 8.90825366973877
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic54-recog_dic54] --> diff 0.0 loss1 8.988865852355957 loss2 8.988865852355957
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic55-recog_dic55] --> diff 0.0 loss1 9.660772323608398 loss2 9.660772323608398
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic56-recog_dic56] --> diff 0.0 loss1 9.300654411315918 loss2 9.300654411315918
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic57-recog_dic57] --> diff 0.0 loss1 8.424280166625977 loss2 8.424280166625977
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic58-recog_dic58] --> diff 0.0 loss1 8.958274841308594 loss2 8.958274841308594
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic59-recog_dic59] --> diff 0.0 loss1 9.306142807006836 loss2 9.306142807006836
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic60-recog_dic60] --> diff 0.0 loss1 8.966218948364258 loss2 8.966218948364258
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic61-recog_dic61] --> diff 0.0 loss1 9.057085037231445 loss2 9.057085037231445
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic62-recog_dic62] --> diff 0.0 loss1 9.597885131835938 loss2 9.597885131835938
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic63-recog_dic63] --> diff 0.0 loss1 10.711788177490234 loss2 10.711788177490234
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic64-recog_dic64] --> diff 0.0 loss1 9.067981719970703 loss2 9.067981719970703
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic65-recog_dic65] --> diff 0.0 loss1 10.214668273925781 loss2 10.214668273925781
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic66-recog_dic66] --> diff 0.0 loss1 9.065681457519531 loss2 9.065681457519531
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic67-recog_dic67] --> diff 0.0 loss1 9.1944580078125 loss2 9.1944580078125
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic68-recog_dic68] --> diff 0.0 loss1 9.53188705444336 loss2 9.53188705444336
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic69-recog_dic69] --> diff 0.0 loss1 10.27214241027832 loss2 10.27214241027832
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic70-recog_dic70] --> diff 0.0 loss1 9.188779830932617 loss2 9.188779830932617
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_gpu_trainable[warp-transducer] --> diff 9.5367431640625e-07 loss1 10.335098266601562 loss2 10.335097312927246
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_transducer_gpu_trainable[warp-rnnt] PASSED
test/test_e2e_asr_transducer.py::test_pytorch_multi_gpu_trainable[train_dic0] --> diff 0.0 loss1 6.778289318084717 loss2 6.778289318084717
--> diff 0.0 loss1 11.308906555175781 loss2 11.308906555175781
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_multi_gpu_trainable[train_dic1] --> diff 0.0 loss1 6.942318916320801 loss2 6.942318916320801
--> diff 0.0 loss1 11.342866897583008 loss2 11.342866897583008
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_calculate_attentions[noatt] (9, 3)
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_calculate_attentions[add] (9, 3)
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_calculate_attentions[location] (9, 3)
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_calculate_attentions[location2d] (9, 3)
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_calculate_attentions[multi_head_dot] (2, 9, 3)
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_calculate_attentions[multi_head_add] (2, 9, 3)
PASSED
test/test_e2e_asr_transducer.py::test_pytorch_calculate_attentions[multi_head_loc] (2, 9, 3)
PASSED

=============================== warnings summary ===============================
../anaconda3/lib/python3.7/site-packages/_pytest/config/__init__.py:1233
  /private/home/vincentqb/anaconda3/lib/python3.7/site-packages/_pytest/config/__init__.py:1233: PytestConfigWarning: Unknown config option: execution_timeout
  
    self._warn_or_fail_if_strict(f"Unknown config option: {key}\n")

test/test_e2e_asr_transducer.py::test_pytorch_multi_gpu_trainable[train_dic0]
  /private/home/vincentqb/anaconda3/lib/python3.7/site-packages/torch/cuda/nccl.py:48: DeprecationWarning: Using or importing the ABCs from 'collections' instead of from 'collections.abc' is deprecated, and in 3.8 it will stop working
    if not isinstance(inputs, collections.Container) or isinstance(inputs, torch.Tensor):

-- Docs: https://docs.pytest.org/en/stable/warnings.html

----------- coverage: platform linux, python 3.7.3-final-0 -----------
Name                                                                   Stmts   Miss  Cover
------------------------------------------------------------------------------------------
espnet/__init__.py                                                         6      2    67%
espnet/asr/__init__.py                                                     0      0   100%
espnet/asr/asr_mix_utils.py                                               86     86     0%
espnet/asr/asr_utils.py                                                  431    431     0%
espnet/asr/chainer_backend/__init__.py                                     0      0   100%
espnet/asr/chainer_backend/asr.py                                        208    208     0%
espnet/asr/pytorch_backend/__init__.py                                     0      0   100%
espnet/asr/pytorch_backend/asr.py                                        599    599     0%
espnet/asr/pytorch_backend/asr_init.py                                   119    119     0%
espnet/asr/pytorch_backend/asr_mix.py                                    250    250     0%
espnet/asr/pytorch_backend/recog.py                                       74     74     0%
espnet/bin/__init__.py                                                     0      0   100%
espnet/bin/asr_align.py                                                  137    137     0%
espnet/bin/asr_enhance.py                                                 61     61     0%
espnet/bin/asr_recog.py                                                  109    109     0%
espnet/bin/asr_train.py                                                  176    161     9%
espnet/bin/lm_train.py                                                    95     95     0%
espnet/bin/mt_train.py                                                   136    136     0%
espnet/bin/mt_trans.py                                                    61     61     0%
espnet/bin/st_train.py                                                   151    151     0%
espnet/bin/st_trans.py                                                    60     60     0%
espnet/bin/tts_decode.py                                                  61     61     0%
espnet/bin/tts_train.py                                                   93     93     0%
espnet/bin/vc_decode.py                                                   60     60     0%
espnet/bin/vc_train.py                                                    94     94     0%
espnet/lm/__init__.py                                                      0      0   100%
espnet/lm/chainer_backend/__init__.py                                      0      0   100%
espnet/lm/chainer_backend/extlm.py                                       133    133     0%
espnet/lm/chainer_backend/lm.py                                          248    248     0%
espnet/lm/lm_utils.py                                                    155    114    26%
espnet/lm/pytorch_backend/__init__.py                                      0      0   100%
espnet/lm/pytorch_backend/extlm.py                                       141     70    50%
espnet/lm/pytorch_backend/lm.py                                          201    201     0%
espnet/mt/__init__.py                                                      0      0   100%
espnet/mt/mt_utils.py                                                     35     35     0%
espnet/mt/pytorch_backend/__init__.py                                      0      0   100%
espnet/mt/pytorch_backend/mt.py                                          215    215     0%
espnet/nets/__init__.py                                                    0      0   100%
espnet/nets/asr_interface.py                                              33     14    58%
espnet/nets/batch_beam_search.py                                          97     97     0%
espnet/nets/beam_search.py                                               169    169     0%
espnet/nets/beam_search_transducer.py                                    287      2    99%
espnet/nets/chainer_backend/__init__.py                                    0      0   100%
espnet/nets/chainer_backend/asr_interface.py                               6      6     0%
espnet/nets/chainer_backend/ctc.py                                        64     64     0%
espnet/nets/chainer_backend/deterministic_embed_id.py                     88     88     0%
espnet/nets/chainer_backend/e2e_asr.py                                    94     94     0%
espnet/nets/chainer_backend/e2e_asr_transformer.py                       298    298     0%
espnet/nets/chainer_backend/nets_utils.py                                  5      5     0%
espnet/nets/chainer_backend/rnn/__init__.py                                0      0   100%
espnet/nets/chainer_backend/rnn/attentions.py                            105    105     0%
espnet/nets/chainer_backend/rnn/decoders.py                              246    246     0%
espnet/nets/chainer_backend/rnn/encoders.py                              117    117     0%
espnet/nets/chainer_backend/rnn/training.py                              106    106     0%
espnet/nets/chainer_backend/transformer/__init__.py                        0      0   100%
espnet/nets/chainer_backend/transformer/attention.py                      38     38     0%
espnet/nets/chainer_backend/transformer/ctc.py                            54     54     0%
espnet/nets/chainer_backend/transformer/decoder.py                        49     49     0%
espnet/nets/chainer_backend/transformer/decoder_layer.py                  28     28     0%
espnet/nets/chainer_backend/transformer/embedding.py                      18     18     0%
espnet/nets/chainer_backend/transformer/encoder.py                        52     52     0%
espnet/nets/chainer_backend/transformer/encoder_layer.py                  24     24     0%
espnet/nets/chainer_backend/transformer/label_smoothing_loss.py           35     35     0%
espnet/nets/chainer_backend/transformer/layer_norm.py                      7      7     0%
espnet/nets/chainer_backend/transformer/mask.py                            6      6     0%
espnet/nets/chainer_backend/transformer/plot.py                           56     56     0%
espnet/nets/chainer_backend/transformer/positionwise_feed_forward.py      19     19     0%
espnet/nets/chainer_backend/transformer/subsampling.py                    45     45     0%
espnet/nets/chainer_backend/transformer/training.py                      133    133     0%
espnet/nets/ctc_prefix_score.py                                          148    137     7%
espnet/nets/e2e_asr_common.py                                            183    146    20%
espnet/nets/e2e_mt_common.py                                              34     34     0%
espnet/nets/lm_interface.py                                               22     11    50%
espnet/nets/mt_interface.py                                               21     21     0%
espnet/nets/pytorch_backend/__init__.py                                    0      0   100%
espnet/nets/pytorch_backend/conformer/__init__.py                          0      0   100%
espnet/nets/pytorch_backend/conformer/argument.py                          9      9     0%
espnet/nets/pytorch_backend/conformer/convolution.py                      19     14    26%
espnet/nets/pytorch_backend/conformer/encoder.py                          74     74     0%
espnet/nets/pytorch_backend/conformer/encoder_layer.py                    75     68     9%
espnet/nets/pytorch_backend/conformer/swish.py                             5      1    80%
espnet/nets/pytorch_backend/ctc.py                                       110     94    15%
espnet/nets/pytorch_backend/e2e_asr.py                                   265    217    18%
espnet/nets/pytorch_backend/e2e_asr_conformer.py                          19     19     0%
espnet/nets/pytorch_backend/e2e_asr_maskctc.py                           120    120     0%
espnet/nets/pytorch_backend/e2e_asr_mix.py                               347    347     0%
espnet/nets/pytorch_backend/e2e_asr_mix_transformer.py                   214    214     0%
espnet/nets/pytorch_backend/e2e_asr_mulenc.py                            322    322     0%
espnet/nets/pytorch_backend/e2e_asr_transducer.py                        188     76    60%
espnet/nets/pytorch_backend/e2e_asr_transformer.py                       266    266     0%
espnet/nets/pytorch_backend/e2e_mt.py                                    167    167     0%
espnet/nets/pytorch_backend/e2e_mt_transformer.py                        198    198     0%
espnet/nets/pytorch_backend/e2e_st.py                                    284    284     0%
espnet/nets/pytorch_backend/e2e_st_conformer.py                           19     19     0%
espnet/nets/pytorch_backend/e2e_st_transformer.py                        249    249     0%
espnet/nets/pytorch_backend/e2e_tts_fastspeech.py                        259    259     0%
espnet/nets/pytorch_backend/e2e_tts_tacotron2.py                         250    250     0%
espnet/nets/pytorch_backend/e2e_tts_transformer.py                       354    354     0%
espnet/nets/pytorch_backend/e2e_vc_tacotron2.py                          245    245     0%
espnet/nets/pytorch_backend/e2e_vc_transformer.py                        350    350     0%
espnet/nets/pytorch_backend/fastspeech/__init__.py                         0      0   100%
espnet/nets/pytorch_backend/fastspeech/duration_calculator.py             31     31     0%
espnet/nets/pytorch_backend/fastspeech/duration_predictor.py              35     35     0%
espnet/nets/pytorch_backend/fastspeech/length_regulator.py                25     25     0%
espnet/nets/pytorch_backend/frontends/__init__.py                          0      0   100%
espnet/nets/pytorch_backend/frontends/beamformer.py                       24     18    25%
espnet/nets/pytorch_backend/frontends/dnn_beamformer.py                   72     55    24%
espnet/nets/pytorch_backend/frontends/dnn_wpe.py                          34     25    26%
espnet/nets/pytorch_backend/frontends/feature_transform.py               102     77    25%
espnet/nets/pytorch_backend/frontends/frontend.py                         51     37    27%
espnet/nets/pytorch_backend/frontends/mask_estimator.py                   38     27    29%
espnet/nets/pytorch_backend/initialization.py                             30      8    73%
espnet/nets/pytorch_backend/lm/__init__.py                                 0      0   100%
espnet/nets/pytorch_backend/lm/default.py                                147     68    54%
espnet/nets/pytorch_backend/lm/seq_rnn.py                                 60     60     0%
espnet/nets/pytorch_backend/lm/transformer.py                             63     63     0%
espnet/nets/pytorch_backend/maskctc/__init__.py                            0      0   100%
espnet/nets/pytorch_backend/maskctc/add_mask_token.py                     13     13     0%
espnet/nets/pytorch_backend/maskctc/mask.py                                7      7     0%
espnet/nets/pytorch_backend/nets_utils.py                                125     73    42%
espnet/nets/pytorch_backend/rnn/__init__.py                                0      0   100%
espnet/nets/pytorch_backend/rnn/argument.py                               25     21    16%
espnet/nets/pytorch_backend/rnn/attentions.py                            749    123    84%
espnet/nets/pytorch_backend/rnn/decoders.py                              580    545     6%
espnet/nets/pytorch_backend/rnn/encoders.py                              145     14    90%
espnet/nets/pytorch_backend/streaming/__init__.py                          0      0   100%
espnet/nets/pytorch_backend/streaming/segment.py                          62     62     0%
espnet/nets/pytorch_backend/streaming/window.py                           36     36     0%
espnet/nets/pytorch_backend/tacotron2/__init__.py                          0      0   100%
espnet/nets/pytorch_backend/tacotron2/cbhg.py                             87     87     0%
espnet/nets/pytorch_backend/tacotron2/decoder.py                         222    222     0%
espnet/nets/pytorch_backend/tacotron2/encoder.py                          51     51     0%
espnet/nets/pytorch_backend/transducer/__init__.py                         0      0   100%
espnet/nets/pytorch_backend/transducer/blocks.py                         157    134    15%
espnet/nets/pytorch_backend/transducer/causal_conv1d.py                   13      9    31%
espnet/nets/pytorch_backend/transducer/initializer.py                     17      6    65%
espnet/nets/pytorch_backend/transducer/joint_network.py                   13      0   100%
espnet/nets/pytorch_backend/transducer/loss.py                            34      5    85%
espnet/nets/pytorch_backend/transducer/rnn_att_decoder.py                136      0   100%
espnet/nets/pytorch_backend/transducer/rnn_decoder.py                    103      0   100%
espnet/nets/pytorch_backend/transducer/tdnn.py                            44     37    16%
espnet/nets/pytorch_backend/transducer/transformer_decoder.py             92     74    20%
espnet/nets/pytorch_backend/transducer/transformer_decoder_layer.py       29     23    21%
espnet/nets/pytorch_backend/transducer/transformer_encoder.py             22     14    36%
espnet/nets/pytorch_backend/transducer/utils.py                           98     32    67%
espnet/nets/pytorch_backend/transducer/vgg2l.py                           24     17    29%
espnet/nets/pytorch_backend/transformer/__init__.py                        0      0   100%
espnet/nets/pytorch_backend/transformer/add_sos_eos.py                    10     10     0%
espnet/nets/pytorch_backend/transformer/argument.py                       23     23     0%
espnet/nets/pytorch_backend/transformer/attention.py                      72     58    19%
espnet/nets/pytorch_backend/transformer/decoder.py                        92     92     0%
espnet/nets/pytorch_backend/transformer/decoder_layer.py                  57     57     0%
espnet/nets/pytorch_backend/transformer/dynamic_conv.py                   55     55     0%
espnet/nets/pytorch_backend/transformer/dynamic_conv2d.py                 64     64     0%
espnet/nets/pytorch_backend/transformer/embedding.py                      54     39    28%
espnet/nets/pytorch_backend/transformer/encoder.py                        97     97     0%
espnet/nets/pytorch_backend/transformer/encoder_layer.py                  42     35    17%
espnet/nets/pytorch_backend/transformer/encoder_mix.py                    45     45     0%
espnet/nets/pytorch_backend/transformer/initializer.py                    23     19    17%
espnet/nets/pytorch_backend/transformer/label_smoothing_loss.py           28     28     0%
espnet/nets/pytorch_backend/transformer/layer_norm.py                     10      5    50%
espnet/nets/pytorch_backend/transformer/lightconv.py                      45     45     0%
espnet/nets/pytorch_backend/transformer/lightconv2d.py                    51     51     0%
espnet/nets/pytorch_backend/transformer/mask.py                           16      8    50%
espnet/nets/pytorch_backend/transformer/multi_layer_conv.py               20     20     0%
espnet/nets/pytorch_backend/transformer/optimizer.py                      35     35     0%
espnet/nets/pytorch_backend/transformer/plot.py                           88     88     0%
espnet/nets/pytorch_backend/transformer/positionwise_feed_forward.py      11      6    45%
espnet/nets/pytorch_backend/transformer/repeat.py                          9      4    56%
espnet/nets/pytorch_backend/transformer/subsampling.py                    45     32    29%
espnet/nets/pytorch_backend/wavenet.py                                   183    183     0%
espnet/nets/scorer_interface.py                                           30     13    57%
espnet/nets/scorers/__init__.py                                            0      0   100%
espnet/nets/scorers/ctc.py                                                38     26    32%
espnet/nets/scorers/length_bonus.py                                       12     12     0%
espnet/nets/scorers/ngram.py                                              31     31     0%
espnet/nets/st_interface.py                                               10     10     0%
espnet/nets/transducer_decoder_interface.py                               13      0   100%
espnet/nets/tts_interface.py                                              25     25     0%
espnet/optimizer/__init__.py                                               0      0   100%
espnet/optimizer/chainer.py                                               38     38     0%
espnet/optimizer/factory.py                                               23     23     0%
espnet/optimizer/parser.py                                                15     15     0%
espnet/optimizer/pytorch.py                                               28     28     0%
espnet/scheduler/__init__.py                                               0      0   100%
espnet/scheduler/chainer.py                                               14     14     0%
espnet/scheduler/pytorch.py                                               14     14     0%
espnet/scheduler/scheduler.py                                             72     72     0%
espnet/st/__init__.py                                                      0      0   100%
espnet/st/pytorch_backend/__init__.py                                      0      0   100%
espnet/st/pytorch_backend/st.py                                          245    245     0%
espnet/transform/__init__.py                                               0      0   100%
espnet/transform/add_deltas.py                                            24     24     0%
espnet/transform/channel_selector.py                                      20     20     0%
espnet/transform/cmvn.py                                                  82     82     0%
espnet/transform/functional.py                                            34     34     0%
espnet/transform/perturb.py                                              178    178     0%
espnet/transform/spec_augment.py                                          91     91     0%
espnet/transform/spectrogram.py                                           93     93     0%
espnet/transform/transform_interface.py                                    9      9     0%
espnet/transform/transformation.py                                        67     67     0%
espnet/transform/wpe.py                                                   13     13     0%
espnet/tts/__init__.py                                                     0      0   100%
espnet/tts/pytorch_backend/__init__.py                                     0      0   100%
espnet/tts/pytorch_backend/tts.py                                        341    341     0%
espnet/utils/__init__.py                                                   0      0   100%
espnet/utils/check_kwargs.py                                              11     11     0%
espnet/utils/cli_readers.py                                              136    136     0%
espnet/utils/cli_utils.py                                                 14      6    57%
espnet/utils/cli_writers.py                                              131    131     0%
espnet/utils/dataset.py                                                   48     48     0%
espnet/utils/deterministic_utils.py                                       25     25     0%
espnet/utils/dynamic_import.py                                             9      7    22%
espnet/utils/fill_missing_args.py                                         13     10    23%
espnet/utils/io_utils.py                                                 297    297     0%
espnet/utils/spec_augment.py                                             183    183     0%
espnet/utils/training/__init__.py                                          0      0   100%
espnet/utils/training/batchfy.py                                         214    204     5%
espnet/utils/training/evaluator.py                                        12     12     0%
espnet/utils/training/iterators.py                                        34     34     0%
espnet/utils/training/tensorboard_logger.py                               26     26     0%
espnet/utils/training/train_utils.py                                      13     13     0%
espnet2/__init__.py                                                        0      0   100%
espnet2/asr/__init__.py                                                    0      0   100%
espnet2/asr/ctc.py                                                        65     65     0%
espnet2/asr/decoder/__init__.py                                            0      0   100%
espnet2/asr/decoder/abs_decoder.py                                         6      0   100%
espnet2/asr/decoder/rnn_decoder.py                                       149    149     0%
espnet2/asr/decoder/transformer_decoder.py                               119    119     0%
espnet2/asr/encoder/__init__.py                                            0      0   100%
espnet2/asr/encoder/abs_encoder.py                                         6      6     0%
espnet2/asr/encoder/conformer_encoder.py                                  86     86     0%
espnet2/asr/encoder/rnn_encoder.py                                        42     42     0%
espnet2/asr/encoder/transformer_encoder.py                                61     61     0%
espnet2/asr/encoder/vgg_rnn_encoder.py                                    39     39     0%
espnet2/asr/espnet_joint_model.py                                        292    292     0%
espnet2/asr/espnet_model.py                                              120    120     0%
espnet2/asr/frontend/__init__.py                                           0      0   100%
espnet2/asr/frontend/abs_frontend.py                                       5      5     0%
espnet2/asr/frontend/default.py                                           54     54     0%
espnet2/asr/frontend/windowing.py                                         27     27     0%
espnet2/asr/preencoder/__init__.py                                         0      0   100%
espnet2/asr/preencoder/abs_preencoder.py                                   5      5     0%
espnet2/asr/preencoder/sinc.py                                            90     90     0%
espnet2/asr/specaug/__init__.py                                            0      0   100%
espnet2/asr/specaug/abs_specaug.py                                         5      5     0%
espnet2/asr/specaug/specaug.py                                            35     35     0%
espnet2/bin/__init__.py                                                    0      0   100%
espnet2/bin/aggregate_stats_dirs.py                                       58     58     0%
espnet2/bin/asr_inference.py                                             173    173     0%
espnet2/bin/asr_train.py                                                   6      6     0%
espnet2/bin/enh_inference.py                                              83     83     0%
espnet2/bin/enh_scoring.py                                                60     60     0%
espnet2/bin/enh_train.py                                                   6      6     0%
espnet2/bin/launch.py                                                    139    139     0%
espnet2/bin/lm_calc_perplexity.py                                         95     95     0%
espnet2/bin/lm_train.py                                                    6      6     0%
espnet2/bin/pack.py                                                       39     39     0%
espnet2/bin/split_scps.py                                                 59     59     0%
espnet2/bin/tokenize_text.py                                             112    112     0%
espnet2/bin/tts_inference.py                                             220    220     0%
espnet2/bin/tts_train.py                                                   6      6     0%
espnet2/enh/__init__.py                                                    0      0   100%
espnet2/enh/abs_enh.py                                                     6      6     0%
espnet2/enh/espnet_model.py                                              180    180     0%
espnet2/enh/layers/__init__.py                                             0      0   100%
espnet2/enh/layers/conv_beamformer.py                                     86     86     0%
espnet2/enh/layers/dnn_beamformer.py                                     134    134     0%
espnet2/enh/layers/dnn_wpe.py                                             41     41     0%
espnet2/enh/layers/mask_estimator.py                                      48     48     0%
espnet2/enh/nets/__init__.py                                               0      0   100%
espnet2/enh/nets/beamformer_net.py                                        85     85     0%
espnet2/enh/nets/tasnet.py                                               200    200     0%
espnet2/enh/nets/tf_mask_net.py                                           54     54     0%
espnet2/fileio/__init__.py                                                 0      0   100%
espnet2/fileio/datadir_writer.py                                          49     49     0%
espnet2/fileio/npy_scp.py                                                 48     48     0%
espnet2/fileio/rand_gen_dataset.py                                        34     34     0%
espnet2/fileio/read_text.py                                               43     43     0%
espnet2/fileio/sound_scp.py                                               63     63     0%
espnet2/iterators/__init__.py                                              0      0   100%
espnet2/iterators/abs_iter_factory.py                                      4      4     0%
espnet2/iterators/chunk_iter_factory.py                                   98     98     0%
espnet2/iterators/multiple_iter_factory.py                                24     24     0%
espnet2/iterators/sequence_iter_factory.py                                74     74     0%
espnet2/layers/__init__.py                                                 0      0   100%
espnet2/layers/abs_normalize.py                                            5      5     0%
espnet2/layers/global_mvn.py                                              68     68     0%
espnet2/layers/inversible_interface.py                                     5      5     0%
espnet2/layers/log_mel.py                                                 30     30     0%
espnet2/layers/mask_along_axis.py                                         55     55     0%
espnet2/layers/sinc_conv.py                                              108    108     0%
espnet2/layers/stft.py                                                    64     64     0%
espnet2/layers/time_warp.py                                               38     38     0%
espnet2/layers/utterance_mvn.py                                           38     38     0%
espnet2/lm/__init__.py                                                     0      0   100%
espnet2/lm/abs_model.py                                                    6      6     0%
espnet2/lm/espnet_model.py                                                39     39     0%
espnet2/lm/seq_rnn_lm.py                                                  59     59     0%
espnet2/lm/transformer_lm.py                                              48     48     0%
espnet2/main_funcs/__init__.py                                             0      0   100%
espnet2/main_funcs/average_nbest_models.py                                55     55     0%
espnet2/main_funcs/calculate_all_attentions.py                            74     74     0%
espnet2/main_funcs/collect_stats.py                                       67     67     0%
espnet2/main_funcs/pack_funcs.py                                         206    206     0%
espnet2/optimizers/__init__.py                                             0      0   100%
espnet2/optimizers/sgd.py                                                  6      6     0%
espnet2/samplers/__init__.py                                               0      0   100%
espnet2/samplers/abs_sampler.py                                            8      8     0%
espnet2/samplers/build_batch_sampler.py                                   32     32     0%
espnet2/samplers/folded_batch_sampler.py                                  86     86     0%
espnet2/samplers/length_batch_sampler.py                                  77     77     0%
espnet2/samplers/num_elements_batch_sampler.py                            84     84     0%
espnet2/samplers/sorted_batch_sampler.py                                  41     41     0%
espnet2/samplers/unsorted_batch_sampler.py                                41     41     0%
espnet2/schedulers/__init__.py                                             0      0   100%
espnet2/schedulers/abs_scheduler.py                                       15     15     0%
espnet2/schedulers/noam_lr.py                                             24     24     0%
espnet2/schedulers/warmup_lr.py                                           17     17     0%
espnet2/tasks/__init__.py                                                  0      0   100%
espnet2/tasks/abs_task.py                                                649    649     0%
espnet2/tasks/asr.py                                                     143    143     0%
espnet2/tasks/enh.py                                                      73     73     0%
espnet2/tasks/enh_asr.py                                                 143    143     0%
espnet2/tasks/lm.py                                                       90     90     0%
espnet2/tasks/tts.py                                                     139    139     0%
espnet2/text/__init__.py                                                   0      0   100%
espnet2/text/abs_tokenizer.py                                              5      5     0%
espnet2/text/build_tokenizer.py                                           26     26     0%
espnet2/text/char_tokenizer.py                                            38     38     0%
espnet2/text/cleaner.py                                                   28     28     0%
espnet2/text/phoneme_tokenizer.py                                         98     98     0%
espnet2/text/sentencepiece_tokenizer.py                                   24     24     0%
espnet2/text/token_id_converter.py                                        42     42     0%
espnet2/text/word_tokenizer.py                                            35     35     0%
espnet2/torch_utils/__init__.py                                            0      0   100%
espnet2/torch_utils/add_gradient_noise.py                                  9      9     0%
espnet2/torch_utils/device_funcs.py                                       39     39     0%
espnet2/torch_utils/forward_adaptor.py                                    13     13     0%
espnet2/torch_utils/initialize.py                                         53     53     0%
espnet2/torch_utils/load_pretrained_model.py                              38     38     0%
espnet2/torch_utils/model_summary.py                                      27     27     0%
espnet2/torch_utils/pytorch_version.py                                     6      6     0%
espnet2/torch_utils/recursive_op.py                                       40     40     0%
espnet2/torch_utils/set_all_random_seed.py                                 7      7     0%
espnet2/train/__init__.py                                                  0      0   100%
espnet2/train/abs_espnet_model.py                                          6      6     0%
espnet2/train/class_choices.py                                            41     41     0%
espnet2/train/collate_fn.py                                               41     41     0%
espnet2/train/dataset.py                                                 193    193     0%
espnet2/train/distributed_utils.py                                       176    176     0%
espnet2/train/iterable_dataset.py                                        125    125     0%
espnet2/train/preprocessor.py                                             69     69     0%
espnet2/train/reporter.py                                                363    363     0%
espnet2/train/trainer.py                                                 307    307     0%
espnet2/tts/__init__.py                                                    0      0   100%
espnet2/tts/abs_tts.py                                                     6      6     0%
espnet2/tts/duration_calculator.py                                        28     28     0%
espnet2/tts/espnet_model.py                                               99     99     0%
espnet2/tts/fastspeech.py                                                147    147     0%
espnet2/tts/fastspeech2.py                                               214    214     0%
espnet2/tts/feats_extract/__init__.py                                      0      0   100%
espnet2/tts/feats_extract/abs_feats_extract.py                             7      7     0%
espnet2/tts/feats_extract/dio.py                                          87     87     0%
espnet2/tts/feats_extract/energy.py                                       62     62     0%
espnet2/tts/feats_extract/log_mel_fbank.py                                39     39     0%
espnet2/tts/feats_extract/log_spectrogram.py                              29     29     0%
espnet2/tts/gst/__init__.py                                                0      0   100%
espnet2/tts/gst/style_encoder.py                                          72     72     0%
espnet2/tts/tacotron2.py                                                 144    144     0%
espnet2/tts/transformer.py                                               238    238     0%
espnet2/tts/variance_predictor.py                                         21     21     0%
espnet2/utils/__init__.py                                                  0      0   100%
espnet2/utils/build_dataclass.py                                          11     11     0%
espnet2/utils/config_argparse.py                                          23     23     0%
espnet2/utils/get_default_kwargs.py                                       34     34     0%
espnet2/utils/griffin_lim.py                                              49     49     0%
espnet2/utils/nested_dict_action.py                                       40     40     0%
espnet2/utils/sized_dict.py                                               46     46     0%
espnet2/utils/types.py                                                    50     50     0%
espnet2/utils/yaml_no_alias_safe_dump.py                                   6      6     0%
------------------------------------------------------------------------------------------
TOTAL                                                                  28260  25965     8%

============================== slowest durations ===============================
4.54s call     test/test_e2e_asr_transducer.py::test_pytorch_multi_gpu_trainable[train_dic0]
4.47s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_gpu_trainable[warp-transducer]
0.09s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic21-recog_dic21]
0.08s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic2-recog_dic2]
0.07s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic36-recog_dic36]
0.07s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic47-recog_dic47]
0.07s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic31-recog_dic31]
0.07s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic37-recog_dic37]
0.07s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic39-recog_dic39]
0.07s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic51-recog_dic51]
0.07s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic62-recog_dic62]
0.07s call     test/test_e2e_asr_transducer.py::test_pytorch_multi_gpu_trainable[train_dic1]
0.07s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic0-recog_dic0]
0.06s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic49-recog_dic49]
0.06s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic8-recog_dic8]
0.06s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic59-recog_dic59]
0.06s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic57-recog_dic57]
0.06s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic22-recog_dic22]
0.06s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic53-recog_dic53]
0.06s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic55-recog_dic55]
0.06s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic70-recog_dic70]
0.06s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic61-recog_dic61]
0.06s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic32-recog_dic32]
0.06s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic33-recog_dic33]
0.06s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic35-recog_dic35]
0.05s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic16-recog_dic16]
0.05s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic41-recog_dic41]
0.05s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic45-recog_dic45]
0.05s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic66-recog_dic66]
0.05s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic6-recog_dic6]
0.05s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic24-recog_dic24]
0.05s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic65-recog_dic65]
0.05s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic14-recog_dic14]
0.05s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic43-recog_dic43]
0.05s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic60-recog_dic60]
0.05s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic4-recog_dic4]
0.05s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic12-recog_dic12]
0.05s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic34-recog_dic34]
0.05s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic26-recog_dic26]
0.05s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic68-recog_dic68]
0.05s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic18-recog_dic18]
0.05s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic69-recog_dic69]
0.05s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic67-recog_dic67]
0.05s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic10-recog_dic10]
0.05s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic29-recog_dic29]
0.05s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic50-recog_dic50]
0.05s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic63-recog_dic63]
0.04s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic46-recog_dic46]
0.04s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic64-recog_dic64]
0.04s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic30-recog_dic30]
0.04s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic40-recog_dic40]
0.04s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic28-recog_dic28]
0.04s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic48-recog_dic48]
0.04s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic56-recog_dic56]
0.04s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic58-recog_dic58]
0.04s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic20-recog_dic20]
0.04s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic3-recog_dic3]
0.04s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic52-recog_dic52]
0.04s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic54-recog_dic54]
0.04s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic7-recog_dic7]
0.04s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic25-recog_dic25]
0.04s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic17-recog_dic17]
0.04s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic42-recog_dic42]
0.04s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic27-recog_dic27]
0.03s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic44-recog_dic44]
0.03s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic15-recog_dic15]
0.03s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic38-recog_dic38]
0.03s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic13-recog_dic13]
0.03s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic1-recog_dic1]
0.03s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic9-recog_dic9]
0.03s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic19-recog_dic19]
0.03s call     test/test_e2e_asr_transducer.py::test_pytorch_calculate_attentions[multi_head_loc]
0.03s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic23-recog_dic23]
0.03s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic11-recog_dic11]
0.03s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_trainable_and_decodable[train_dic5-recog_dic5]
0.03s call     test/test_e2e_asr_transducer.py::test_pytorch_calculate_attentions[noatt]
0.03s call     test/test_e2e_asr_transducer.py::test_pytorch_calculate_attentions[multi_head_add]
0.02s call     test/test_e2e_asr_transducer.py::test_pytorch_calculate_attentions[location2d]
0.02s call     test/test_e2e_asr_transducer.py::test_pytorch_calculate_attentions[location]
0.02s call     test/test_e2e_asr_transducer.py::test_pytorch_calculate_attentions[multi_head_dot]
0.02s call     test/test_e2e_asr_transducer.py::test_pytorch_calculate_attentions[add]
0.01s call     test/test_e2e_asr_transducer.py::test_pytorch_transducer_gpu_trainable[warp-rnnt]

(164 durations < 0.005s hidden.  Use -vv to show these durations.)
======================= 82 passed, 2 warnings in 33.51s ========================

@vincentqb
Copy link
Owner Author

Test could also be ran with torchaudio here.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant