Deprecates the use_batch_norm_* options to LSTM. Please switch to Bat…

…chNormLSTM if you require these options, in a future version they will no longer be supported by LSTM. PiperOrigin-RevId: 159420083
google-deepmind · Jun 21, 2017 · 93d5471 · 93d5471
1 parent 0944fc6
commit 93d5471
Show file tree

Hide file tree

Showing 3 changed files with 414 additions and 220 deletions.
diff --git a/sonnet/__init__.py b/sonnet/__init__.py
@@ -76,6 +76,7 @@
 from sonnet.python.modules.conv import SeparableConv2D
 from sonnet.python.modules.conv import VALID
 from sonnet.python.modules.embed import Embed
+from sonnet.python.modules.gated_rnn import BatchNormLSTM
 from sonnet.python.modules.gated_rnn import Conv1DLSTM
 from sonnet.python.modules.gated_rnn import Conv2DLSTM
 from sonnet.python.modules.gated_rnn import GRU