Wei Kang 7012fd65b5

Support mix precision training on the reworked model (#305) ...

* Add mix precision support

* Minor fixes

* Minor fixes

* Minor fixes

2022-04-11 16:49:54 +08:00

..

__init__.py

Start adding some files..

2022-03-16 22:14:30 +08:00

asr_datamodule.py

Start adding some files..

2022-03-16 22:14:30 +08:00

beam_search.py

Modify beam search to be efficient with current joienr

2022-04-11 12:35:29 +08:00

conformer.py

Remove initial_speed

2022-04-06 13:17:26 +08:00

decode.py

Fix dir names

2022-04-11 12:03:41 +08:00

decoder.py

Remove initial_speed

2022-04-06 13:17:26 +08:00

encoder_interface.py

Add remaining files in pruned_transducer_stateless2

2022-03-17 11:17:52 +08:00

export.py

Fix dir names

2022-04-11 12:03:41 +08:00

joiner.py

Remove initial_speed

2022-04-06 13:17:26 +08:00

model.py

Support mix precision training on the reworked model (#305)

2022-04-11 16:49:54 +08:00

optim.py

Fix docs in optim.py

2022-04-11 15:13:42 +08:00

scaling.py

Changing initial_speed from 0.25 to 01

2022-04-05 18:00:54 +08:00

train.py

Support mix precision training on the reworked model (#305)

2022-04-11 16:49:54 +08:00