Merge of PR google-research#57

PiperOrigin-RevId: 310374598
Xtuden-com · May 7, 2020 · 58f5dc3 · 58f5dc3
1 parent a556016
commit 58f5dc3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/language/bert_extraction/steal_bert_qa/scripts/train_victim_squad.sh b/language/bert_extraction/steal_bert_qa/scripts/train_victim_squad.sh
@@ -54,4 +54,4 @@ python -m language.bert_extraction.steal_bert_qa.models.run_squad \
 # For SQuAD 2.0, use the script language.bert_extraction.steal_bert_qa.utils.evaluate_squad_2
 python -m language.bert_extraction.steal_bert_qa.utils.evaluate_squad \
   --dataset_file=$SQUAD_DIR/dev-v1.1.json \
-  --predictions_file=$OUTPUT_DIR/predictions.json
+  --prediction_file=$OUTPUT_DIR/predictions.json
diff --git a/language/bert_extraction/steal_bert_qa/utils/evaluate_squad.py b/language/bert_extraction/steal_bert_qa/utils/evaluate_squad.py
@@ -22,7 +22,7 @@
 import string
 import tensorflow.compat.v1 as tf
 
-app = tf.compat.v1.app
+app = tf.app
 flags = tf.flags
 gfile = tf.gfile
 logging = tf.logging