Unrol RNNs

karibbov · Jan 4, 2021 · aac6f53 · aac6f53
1 parent acf0dd5
commit aac6f53
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorflow_asr/models/transducer.py b/tensorflow_asr/models/transducer.py
@@ -56,7 +56,7 @@ def __init__(self,
                 name=f"{name}_{rnn_type}_{i}", return_state=True,
                 implementation=rnn_implementation,
                 kernel_regularizer=kernel_regularizer,
-                bias_regularizer=bias_regularizer
+                bias_regularizer=bias_regularizer, unroll=True
             )
             if layer_norm:
                 ln = tf.keras.layers.LayerNormalization(name=f"{name}_ln_{i}")