Skip to content

GitLab

Commit 601daf54 authored by xinliupitt's avatar xinliupitt
Browse files

initializers.serialize

parent 9575381a
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/nlp/modeling/layers/transformer.py
View file @ 601daf54
...@@ -223,7 +223,7 @@ class Transformer(tf.keras.layers.Layer): ...@@ -223,7 +223,7 @@ class Transformer(tf.keras.layers.Layer):
"intermediate_dropout": "intermediate_dropout":
self._intermediate_dropout, self._intermediate_dropout,
"attention_initializer": "attention_initializer":
tf.keras.constraints.serialize(self._attention_initializer) tf.keras.initializers.serialize(self._attention_initializer)
} }
base_config = super(Transformer, self).get_config() base_config = super(Transformer, self).get_config()
return dict(list(base_config.items()) + list(config.items())) return dict(list(base_config.items()) + list(config.items()))
...@@ -486,7 +486,7 @@ class TransformerDecoderLayer(tf.keras.layers.Layer): ...@@ -486,7 +486,7 @@ class TransformerDecoderLayer(tf.keras.layers.Layer):
"intermediate_dropout": "intermediate_dropout":
self._intermediate_dropout, self._intermediate_dropout,
"attention_initializer": "attention_initializer":
tf.keras.constraints.serialize(self._attention_initializer) tf.keras.initializers.serialize(self._attention_initializer)
} }
base_config = super(TransformerDecoderLayer, self).get_config() base_config = super(TransformerDecoderLayer, 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