Skip to content

GitLab

Commit fba8ea2f authored by Le Hou's avatar Le Hou Committed by A. Unique TensorFlower
Browse files

Minor bug fixes 

PiperOrigin-RevId: 335944731
parent bc1cc8bb
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
official/nlp/modeling/layers/transformer_scaffold.py
View file @ fba8ea2f
...@@ -224,6 +224,8 @@ class TransformerScaffold(tf.keras.layers.Layer): ...@@ -224,6 +224,8 @@ class TransformerScaffold(tf.keras.layers.Layer):
self._dropout_rate, self._dropout_rate,
"attention_dropout_rate": "attention_dropout_rate":
self._attention_dropout_rate, self._attention_dropout_rate,
"norm_first":
self._norm_first,
"kernel_initializer": "kernel_initializer":
tf.keras.initializers.serialize(self._kernel_initializer), tf.keras.initializers.serialize(self._kernel_initializer),
"bias_initializer": "bias_initializer":
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment