Internal change

PiperOrigin-RevId: 297247401

Internal change
PiperOrigin-RevId: 297247401
7b9365dd · Hongkun Yu · A. Unique TensorFlower · 37c6feb0 · 7b9365dd
Commit 7b9365dd authored Feb 25, 2020 by Hongkun Yu Committed by A. Unique TensorFlower Feb 25, 2020
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/nlp/bert/tf2_encoder_checkpoint_converter.py official/nlp/bert/tf2_encoder_checkpoint_converter.py +1 -1

No files found.
--- a/official/nlp/bert/tf2_encoder_checkpoint_converter.py
+++ b/official/nlp/bert/tf2_encoder_checkpoint_converter.py
@@ -50,7 +50,7 @@ def _create_bert_model(cfg):
  Args:
    cfg: A `BertConfig` to create the core model.
  Returns:
-    A keras model.
+    A TransformerEncoder netowork.
  """
  bert_encoder = networks.TransformerEncoder(
      vocab_size=cfg.vocab_size,