Skip to content

GitLab

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

Internal change 

PiperOrigin-RevId: 328595044
parent c814f04d
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/nlp/modeling/networks/mobile_bert_encoder.py
View file @ 67efd3ab
...@@ -480,7 +480,7 @@ class MobileBERTEncoder(tf.keras.Model): ...@@ -480,7 +480,7 @@ class MobileBERTEncoder(tf.keras.Model):
input_mask = tf.keras.layers.Input( input_mask = tf.keras.layers.Input(
shape=(None,), dtype=tf.int32, name='input_mask') shape=(None,), dtype=tf.int32, name='input_mask')
type_ids = tf.keras.layers.Input( type_ids = tf.keras.layers.Input(
shape=(None,), dtype=tf.int32, name='input_token_type_ids') shape=(None,), dtype=tf.int32, name='input_type_ids')
self.inputs = [input_ids, input_mask, type_ids] self.inputs = [input_ids, input_mask, type_ids]
attention_mask = layers.SelfAttentionMask()([input_ids, input_mask]) attention_mask = layers.SelfAttentionMask()([input_ids, input_mask])
......
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