Skip to content

GitLab

"git@developer.sourcefind.cn:modelzoo/chatglm.git" did not exist on "e07b8855d4f652a065c3c28374d3b1360dd26686"
Commit 7e4837c2 authored by xinliupitt's avatar xinliupitt
Browse files

indent

parent 37e7a323
Hide whitespace changes
Inline Side-by-side
Showing with 36 additions and 36 deletions
official/nlp/modeling/models/seq2seq_transformer.py
View file @ 7e4837c2
...@@ -432,24 +432,24 @@ class TransformerEncoder(tf.keras.layers.Layer): ...@@ -432,24 +432,24 @@ class TransformerEncoder(tf.keras.layers.Layer):
return { return {
"num_layers": "num_layers":
self._num_layers, self._num_layers,
"num_attention_heads": "num_attention_heads":
self._num_attention_heads, self._num_attention_heads,
"intermediate_size": "intermediate_size":
self._intermediate_size, self._intermediate_size,
"activation": "activation":
self._activation, self._activation,
"dropout_rate": "dropout_rate":
self._dropout_rate, self._dropout_rate,
"attention_dropout_rate": "attention_dropout_rate":
self._attention_dropout_rate, self._attention_dropout_rate,
"use_bias": "use_bias":
self._use_bias, self._use_bias,
"norm_first": "norm_first":
self._norm_first, self._norm_first,
"norm_epsilon": "norm_epsilon":
self._norm_epsilon, self._norm_epsilon,
"intermediate_dropout": "intermediate_dropout":
self._intermediate_dropout self._intermediate_dropout
} }
def call(self, def call(self,
...@@ -547,24 +547,24 @@ class TransformerDecoder(tf.keras.layers.Layer): ...@@ -547,24 +547,24 @@ class TransformerDecoder(tf.keras.layers.Layer):
return { return {
"num_layers": "num_layers":
self._num_layers, self._num_layers,
"num_attention_heads": "num_attention_heads":
self._num_attention_heads, self._num_attention_heads,
"intermediate_size": "intermediate_size":
self._intermediate_size, self._intermediate_size,
"activation": "activation":
self._activation, self._activation,
"dropout_rate": "dropout_rate":
self._dropout_rate, self._dropout_rate,
"attention_dropout_rate": "attention_dropout_rate":
self._attention_dropout_rate, self._attention_dropout_rate,
"use_bias": "use_bias":
self._use_bias, self._use_bias,
"norm_first": "norm_first":
self._norm_first, self._norm_first,
"norm_epsilon": "norm_epsilon":
self._norm_epsilon, self._norm_epsilon,
"intermediate_dropout": "intermediate_dropout":
self._intermediate_dropout self._intermediate_dropout
} }
def call(self, def call(self,
......
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