icefall/optim.py at e22bc78f9827ce4059cd4598c19ad08415802c0a - icefall - Bi Git

mirrors/icefall

mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-08 09:32:20 +00:00

Zengwei Yao f3ad32777a

Gradient filter for training lstm model (#564 )

* init files

* add gradient filter module

* refact getting median value

* add cutoff for grad filter

* delete comments

* apply gradient filter in LSTM module, to filter both input and params

* fix typing and refactor

* filter with soft mask

* rename lstm_transducer_stateless2 to lstm_transducer_stateless3

* fix typos, and update RESULTS.md

* minor fix

* fix return typing

* fix typo

2022-09-29 11:15:43 +08:00

Symbolic link

1 line

40 B

Python

Raw Blame History

../pruned_transducer_stateless2/optim.py