Fangjun Kuang 896993714b Add tests to ensure that the model is torch scriptable.

2022-05-16 19:12:34 +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

Add tests to ensure that the model is torch scriptable.

2022-05-16 19:12:34 +08:00

train.py

Keep model_avg on cpu (#348)

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