Skip to content

GitLab

Commit e04eb07b authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 373866819
parent 39464225
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/nlp/modeling/layers/transformer.py
View file @ e04eb07b
...@@ -107,7 +107,7 @@ class Transformer(keras_nlp.layers.TransformerEncoderBlock): ...@@ -107,7 +107,7 @@ class Transformer(keras_nlp.layers.TransformerEncoderBlock):
"intermediate_activation": "intermediate_activation":
self._inner_activation, self._inner_activation,
"dropout_rate": "dropout_rate":
self._attention_dropout_rate, self._output_dropout_rate,
"attention_dropout_rate": "attention_dropout_rate":
self._attention_dropout_rate, self._attention_dropout_rate,
"output_range": "output_range":
......
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