Skip to content

GitLab

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

write_meta_graph=False as we do not care the graph. 

PiperOrigin-RevId: 314394246
parent 8eb91073
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/nlp/bert/tf1_checkpoint_converter_lib.py
View file @ 7580f0a2
...@@ -188,7 +188,7 @@ def convert(checkpoint_from_path, ...@@ -188,7 +188,7 @@ def convert(checkpoint_from_path,
with tf.Session() as sess: with tf.Session() as sess:
sess.run(tf.global_variables_initializer()) sess.run(tf.global_variables_initializer())
tf.logging.info("Writing checkpoint_to_path %s", checkpoint_to_path) tf.logging.info("Writing checkpoint_to_path %s", checkpoint_to_path)
saver.save(sess, checkpoint_to_path) saver.save(sess, checkpoint_to_path, write_meta_graph=False)
tf.logging.info("Summary:") tf.logging.info("Summary:")
tf.logging.info(" Converted %d variable name(s).", len(new_variable_map)) tf.logging.info(" Converted %d variable name(s).", len(new_variable_map))
......
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