Daniel Povey 86c2c60100 Step lr_scheduler on tokens not epoch; add some more debug output

2023-05-04 15:35:22 +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

separate Conv2dSubsampling from Zipformer

2023-04-27 10:11:47 +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

Fix bug with indicator

2023-05-02 13:36:03 +08:00

optim.py

Step lr_scheduler on tokens not epoch; add some more debug output

2023-05-04 15:35:22 +08:00

scaling.py

Add (back) straight_through_rate, with rate 0.025; try to handle memory allocation failures in backprop better.

2023-04-30 15:19:34 +08:00

subsampling.py

separate Conv2dSubsampling from Zipformer

2023-04-27 10:11:47 +08:00

test_model.py

Merge changes from pruned_transducer_stateless4->5

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

train.py

Add text embeddings, but use actual text for now

2023-05-01 22:09:27 +08:00

zipformer.py

Initial version of zipformer1 LM that runs, not sure whether it is working

2023-05-04 14:46:06 +08:00