"examples/vscode:/vscode.git/clone" did not exist on "ea8b93f9b267ca15e3cb2b639470b36aa5a30911"
Commit 295b3b4f authored by Le Hou's avatar Le Hou Committed by A. Unique TensorFlower
Browse files

Fix get_config() in ReZeroTransformer.

PiperOrigin-RevId: 307297217
parent 4e434726
...@@ -187,6 +187,8 @@ class ReZeroTransformer(tf.keras.layers.Layer): ...@@ -187,6 +187,8 @@ class ReZeroTransformer(tf.keras.layers.Layer):
self._dropout_rate, self._dropout_rate,
"attention_dropout_rate": "attention_dropout_rate":
self._attention_dropout_rate, self._attention_dropout_rate,
"use_layer_norm":
self._use_layer_norm,
"kernel_initializer": "kernel_initializer":
tf.keras.initializers.serialize(self._kernel_initializer), tf.keras.initializers.serialize(self._kernel_initializer),
"bias_initializer": "bias_initializer":
...@@ -201,8 +203,6 @@ class ReZeroTransformer(tf.keras.layers.Layer): ...@@ -201,8 +203,6 @@ class ReZeroTransformer(tf.keras.layers.Layer):
tf.keras.constraints.serialize(self._kernel_constraint), tf.keras.constraints.serialize(self._kernel_constraint),
"bias_constraint": "bias_constraint":
tf.keras.constraints.serialize(self._bias_constraint), tf.keras.constraints.serialize(self._bias_constraint),
"use_layer_norm":
tf.keras.constraints.serialize(self._use_layer_norm)
} }
base_config = super(ReZeroTransformer, self).get_config() base_config = super(ReZeroTransformer, self).get_config()
return dict(list(base_config.items()) + list(config.items())) return dict(list(base_config.items()) + list(config.items()))
......
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