Makes do_lower_case not trainable.

PiperOrigin-RevId: 275192365

Makes do_lower_case not trainable.
PiperOrigin-RevId: 275192365
eabff6f0 · Hongkun Yu · A. Unique TensorFlower · 9317f3b4 · eabff6f0
Commit eabff6f0 authored Oct 16, 2019 by Hongkun Yu Committed by A. Unique TensorFlower Oct 16, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

official/nlp/bert/export_tfhub.py official/nlp/bert/export_tfhub.py +2 -1

No files found.
--- a/official/nlp/bert/export_tfhub.py
+++ b/official/nlp/bert/export_tfhub.py
@@ -70,7 +70,8 @@ def export_bert_tfhub(bert_config: bert_modeling.BertConfig,
  checkpoint = tf.train.Checkpoint(model=core_model)
  checkpoint.restore(model_checkpoint_path).assert_consumed()
  core_model.vocab_file = tf.saved_model.Asset(vocab_file)
-  core_model.do_lower_case = tf.Variable("uncased" in vocab_file)
+  core_model.do_lower_case = tf.Variable(
+      "uncased" in vocab_file, trainable=False)
  core_model.save(hub_destination, include_optimizer=False, save_format="tf")