Guo Liyong 9d48f1ce7d train/decode with codebook loss

2022-04-27 14:19:10 +08:00

..

__init__.py

a copy from pruned_transducer_stateless2

2022-04-27 10:55:54 +08:00

asr_datamodule.py

a copy from pruned_transducer_stateless2

2022-04-27 10:55:54 +08:00

beam_search.py

a copy from pruned_transducer_stateless2

2022-04-27 10:55:54 +08:00

checkpoint.py

add codebook loss

2022-04-27 14:17:44 +08:00

conformer.py

middle layer outputs from encoder

2022-04-27 12:33:54 +08:00

decode.py

train/decode with codebook loss

2022-04-27 14:19:10 +08:00

decoder.py

a copy from pruned_transducer_stateless2

2022-04-27 10:55:54 +08:00

encoder_interface.py

a copy from pruned_transducer_stateless2

2022-04-27 10:55:54 +08:00

export.py

a copy from pruned_transducer_stateless2

2022-04-27 10:55:54 +08:00

joiner.py

a copy from pruned_transducer_stateless2

2022-04-27 10:55:54 +08:00

model.py

add codebook loss

2022-04-27 14:17:44 +08:00

optim.py

a copy from pruned_transducer_stateless2

2022-04-27 10:55:54 +08:00

scaling.py

a copy from pruned_transducer_stateless2

2022-04-27 10:55:54 +08:00

train.py

train/decode with codebook loss

2022-04-27 14:19:10 +08:00

vq_prediction.py

add codebook loss

2022-04-27 14:17:44 +08:00