Fangjun Kuang f6ce135608

Various fixes to support torch script. (#371) ...

* Various fixes to support torch script.

* Add tests to ensure that the model is torch scriptable.

* Update tests.

2022-05-16 21:46:59 +08:00

..

__init__.py

Model average (#344)

2022-05-05 21:20:04 +08:00

asr_datamodule.py

Model average (#344)

2022-05-05 21:20:04 +08:00

beam_search.py

Model average (#344)

2022-05-05 21:20:04 +08:00

conformer.py

Model average (#344)

2022-05-05 21:20:04 +08:00

decode.py

Ignore padding frames during RNN-T decoding. (#358)

2022-05-13 07:39:14 +08:00

decoder.py

Model average (#344)

2022-05-05 21:20:04 +08:00

encoder_interface.py

Model average (#344)

2022-05-05 21:20:04 +08:00

export.py

Model average (#344)

2022-05-05 21:20:04 +08:00

joiner.py

Model average (#344)

2022-05-05 21:20:04 +08:00

model.py

Model average (#344)

2022-05-05 21:20:04 +08:00

optim.py

Model average (#344)

2022-05-05 21:20:04 +08:00

scaling.py

Model average (#344)

2022-05-05 21:20:04 +08:00

test_model.py

Various fixes to support torch script. (#371)

2022-05-16 21:46:59 +08:00

train.py

Keep model_avg on cpu (#348)

2022-05-07 10:42:34 +08:00