Daniel Povey e692e0b228 Add balancer for keys

2023-03-07 17:39:01 +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

Remove some unhelpful or unused options in decode.py, setting equivalent to --left-context=0

2023-02-13 12:58:33 +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

Remove some more unused code; rename BasicNorm->BiasNorm, Zipformer->Zipformer2

2023-03-06 14:27:11 +08:00

test_model.py

Merge changes from pruned_transducer_stateless4->5

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

train.py

Remove some more unused code; rename BasicNorm->BiasNorm, Zipformer->Zipformer2

2023-03-06 14:27:11 +08:00

zipformer.py

Add balancer for keys

2023-03-07 17:39:01 +08:00