marcoyang1998 1f0408b103

Support Transformer LM (#750) ...

* support transformer LM

* show number of parameters during training

* update docstring

* testing files for ppl calculation

* add lm wrampper for rnn and transformer LM

* apply lm wrapper in lm shallow fusion

* small updates

* update decode.py to support LM fusion and LODR

* add export.py

* update CI and workflow

* update decoding results

* fix CI

* remove transformer LM from CI test

2022-12-29 10:53:36 +08:00

..

__init__.py

Add LSTM for the multi-dataset setup. (#558)

2022-09-16 18:40:25 +08:00

asr_datamodule.py

Add LSTM for the multi-dataset setup. (#558)

2022-09-16 18:40:25 +08:00

beam_search.py

Add LSTM for the multi-dataset setup. (#558)

2022-09-16 18:40:25 +08:00

decode.py

Support Transformer LM (#750)

2022-12-29 10:53:36 +08:00

decoder.py

Add LSTM for the multi-dataset setup. (#558)

2022-09-16 18:40:25 +08:00

encoder_interface.py

Add LSTM for the multi-dataset setup. (#558)

2022-09-16 18:40:25 +08:00

export.py

manual correction of black formatting

2022-11-17 14:18:05 -05:00

gigaspeech.py

Add LSTM for the multi-dataset setup. (#558)

2022-09-16 18:40:25 +08:00

jit_pretrained.py

manual correction of black formatting

2022-11-17 14:18:05 -05:00

joiner.py

Add LSTM for the multi-dataset setup. (#558)

2022-09-16 18:40:25 +08:00

librispeech.py

Add LSTM for the multi-dataset setup. (#558)

2022-09-16 18:40:25 +08:00

lstm.py

Add LSTM for the multi-dataset setup. (#558)

2022-09-16 18:40:25 +08:00

lstmp.py

Support exporting LSTM with projection to ONNX (#621)

2022-10-18 11:25:31 +08:00

model.py

apply black on all files

2022-11-17 09:42:17 -05:00

ncnn-decode.py

manual correction of black formatting

2022-11-17 14:18:05 -05:00

optim.py

Add LSTM for the multi-dataset setup. (#558)

2022-09-16 18:40:25 +08:00

pretrained.py

manual correction of black formatting

2022-11-17 14:18:05 -05:00

scaling_converter.py

Add LSTM for the multi-dataset setup. (#558)

2022-09-16 18:40:25 +08:00

scaling.py

Add LSTM for the multi-dataset setup. (#558)

2022-09-16 18:40:25 +08:00

streaming-ncnn-decode.py

manual correction of black formatting

2022-11-17 14:18:05 -05:00

streaming-onnx-decode.py

manual correction of black formatting

2022-11-17 14:18:05 -05:00

test_lstmp.py

Support exporting LSTM with projection to ONNX (#621)

2022-10-18 11:25:31 +08:00

train.py

shuffle full Librispeech data (#574)

2022-11-27 11:26:09 +08:00