Skip to content

GitLab

Commit a241b9ae authored by Hongkun Yu's avatar Hongkun Yu Committed by A. Unique TensorFlower
Browse files

Remove @tf.keras.utils.register_keras_serializable(package="Text") as the... 

Remove @tf.keras.utils.register_keras_serializable(package="Text") as the model cannot be serialized anyway.

PiperOrigin-RevId: 408904971
parent 3c72f65f
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
official/nlp/modeling/models/seq2seq_transformer.py
View file @ a241b9ae
...@@ -26,7 +26,6 @@ from official.nlp.modeling.ops import beam_search ...@@ -26,7 +26,6 @@ from official.nlp.modeling.ops import beam_search
EOS_ID = 1 EOS_ID = 1
@tf.keras.utils.register_keras_serializable(package="Text")
class Seq2SeqTransformer(tf.keras.Model): class Seq2SeqTransformer(tf.keras.Model):
"""Transformer model with Keras. """Transformer model with Keras.
......
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