pkufool 2ff81b2838 support half precision training

2022-03-31 19:58:57 +08:00

..

__init__.py

Use k2 pruned transducer loss to train conformer-transducer model (#194)

2022-02-17 13:33:54 +08:00

asr_datamodule.py

Use k2 pruned transducer loss to train conformer-transducer model (#194)

2022-02-17 13:33:54 +08:00

beam_search.py

Batch decoding for models trained with optimized_transducer (#267)

2022-03-23 19:11:34 +08:00

conformer.py

Use k2 pruned transducer loss to train conformer-transducer model (#194)

2022-02-17 13:33:54 +08:00

decode.py

Batch decoding for models trained with optimized_transducer (#267)

2022-03-23 19:11:34 +08:00

decoder.py

Add fast beam search decoding (#250)

2022-03-21 16:22:25 +08:00

encoder_interface.py

Use k2 pruned transducer loss to train conformer-transducer model (#194)

2022-02-17 13:33:54 +08:00

export.py

Add modified beam search for pruned rnn-t. (#248)

2022-03-12 16:16:55 +08:00

joiner.py

Use k2 pruned transducer loss to train conformer-transducer model (#194)

2022-02-17 13:33:54 +08:00

model.py

Use k2 pruned transducer loss to train conformer-transducer model (#194)

2022-02-17 13:33:54 +08:00

pretrained.py

Batch decoding for models trained with optimized_transducer (#267)

2022-03-23 19:11:34 +08:00

subsampling.py

Use k2 pruned transducer loss to train conformer-transducer model (#194)

2022-02-17 13:33:54 +08:00

test_decoder.py

Use k2 pruned transducer loss to train conformer-transducer model (#194)

2022-02-17 13:33:54 +08:00

train.py

support half precision training

2022-03-31 19:58:57 +08:00

transformer.py

Use k2 pruned transducer loss to train conformer-transducer model (#194)

2022-02-17 13:33:54 +08:00