4 Commits

Author SHA1 Message Date
Fangjun Kuang
164321c79d Minor fixes to make it ready for merge. 2021-12-17 19:43:04 +08:00
Fangjun Kuang
fcc22d3e91 Use LSTM layers for the encoder.
Need more tunings.
2021-12-17 11:58:30 +08:00
Fangjun Kuang
f5199d37c4 Use conformer/transformer model as encoder. 2021-12-07 23:20:59 +08:00
Fangjun Kuang
1aff64b708
Apply layer normalization to the output of each gate in LSTM/GRU. (#139)
* Apply layer normalization to the output of each gate in LSTM.

* Apply layer normalization to the output of each gate in GRU.

* Add projection support to LayerNormLSTMCell.

* Add GPU tests.

* Use typeguard.check_argument_types() to validate type annotations.

* Add typeguard as a requirement.

* Minor fixes.

* Fix CI.

* Fix CI.

* Fix test failures for torch 1.8.0

* Fix errors.
2021-12-07 18:38:03 +08:00