Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/pytorch3d.git" did not exist on "327868b86e0d69a3af01299bc5371a56b9ef2ed9"
Commit d4557be2 authored by Le Hou's avatar Le Hou Committed by A. Unique TensorFlower
Browse files

Minor bug fixes 

PiperOrigin-RevId: 335944731
parent 32d5a722
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
official/nlp/modeling/layers/transformer_scaffold.py
View file @ d4557be2
...@@ -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