Skip to content

GitLab

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

Internal change 

PiperOrigin-RevId: 297247401
parent 37c6feb0
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/nlp/bert/tf2_encoder_checkpoint_converter.py
View file @ 7b9365dd
...@@ -50,7 +50,7 @@ def _create_bert_model(cfg): ...@@ -50,7 +50,7 @@ def _create_bert_model(cfg):
Args: Args:
cfg: A `BertConfig` to create the core model. cfg: A `BertConfig` to create the core model.
Returns: Returns:
A keras model. A TransformerEncoder netowork.
""" """
bert_encoder = networks.TransformerEncoder( bert_encoder = networks.TransformerEncoder(
vocab_size=cfg.vocab_size, vocab_size=cfg.vocab_size,
......
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