Skip to content

GitLab

Commit 381dd9fe authored by Chen Chen's avatar Chen Chen Committed by A. Unique TensorFlower
Browse files

Internal Change 

PiperOrigin-RevId: 327094300
parent 9236fd88
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/nlp/bert/tf2_encoder_checkpoint_converter.py
View file @ 381dd9fe
...@@ -42,7 +42,7 @@ flags.DEFINE_string( ...@@ -42,7 +42,7 @@ flags.DEFINE_string(
"BertModel, with no task heads.)") "BertModel, with no task heads.)")
flags.DEFINE_string("converted_checkpoint_path", None, flags.DEFINE_string("converted_checkpoint_path", None,
"Name for the created object-based V2 checkpoint.") "Name for the created object-based V2 checkpoint.")
flags.DEFINE_string("checkpoint_model_name", "model", flags.DEFINE_string("checkpoint_model_name", "encoder",
"The name of the model when saving the checkpoint, i.e., " "The name of the model when saving the checkpoint, i.e., "
"the checkpoint will be saved using: " "the checkpoint will be saved using: "
"tf.train.Checkpoint(FLAGS.checkpoint_model_name=model).") "tf.train.Checkpoint(FLAGS.checkpoint_model_name=model).")
......
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