Daniel Povey dc481ca419 Disable causal training; add balancers in decoder.

2023-02-11 23:10:21 +08:00

..

__init__.py

Merge changes from pruned_transducer_stateless4->5

2022-05-31 13:00:11 +08:00

asr_datamodule.py

Move pruned_transducer_stateless4 to pruned_transducer_stateless7

2022-05-31 12:45:28 +08:00

beam_search.py

Move pruned_transducer_stateless4 to pruned_transducer_stateless7

2022-05-31 12:45:28 +08:00

decode.py

Fix issue in decode.py

2023-02-11 14:45:45 +08:00

decoder.py

Disable causal training; add balancers in decoder.

2023-02-11 23:10:21 +08:00

encoder_interface.py

Move pruned_transducer_stateless4 to pruned_transducer_stateless7

2022-05-31 12:45:28 +08:00

export.py

Use decompose=True arg for model averaging

2022-05-31 14:28:53 +08:00

generate_averaged_model.py

Disable causal training; add balancers in decoder.

2023-02-11 23:10:21 +08:00

joiner.py

Adjust joiner and simple_lm/simple_am projections to account for larger activation dims

2022-12-29 12:52:11 +08:00

model.py

Change how chunk-size is specified

2023-02-11 14:35:31 +08:00

optim.py

Use get_parameter_groups_with_lr in train.py; bug fixes

2023-01-05 14:11:33 +08:00

scaling.py

Cosmetic fix to formula

2023-02-11 15:38:03 +08:00

test_model.py

Merge changes from pruned_transducer_stateless4->5

2022-05-31 13:00:11 +08:00

train.py

Disable causal training; add balancers in decoder.

2023-02-11 23:10:21 +08:00

zipformer.py

Revert warmup_batches change; make code change to avoid non in attn_weights

2023-02-11 18:46:05 +08:00