bug fixes

kamalkraj · Nov 5, 2019 · 00c5eae · 00c5eae
1 parent 6db4d13
commit 00c5eae
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/README.md b/README.md
@@ -133,7 +133,7 @@ python run_squad.py \
   --predict_batch_size=48 \
   --learning_rate=1e-5 \
   --num_train_epochs=3 \
-  --model_dir=${MODEL_DIR} \
+  --model_dir=${OUTPUT_DIR} \
   --strategy_type=mirror
 ```
 

diff --git a/run_squad.py b/run_squad.py
@@ -56,8 +56,6 @@
 flags.DEFINE_integer('predict_batch_size', 8,
                      'Total batch size for predicting.')
 
-flags.DEFINE_integer('predict_batch_size', 8,
-                     'Total batch size for prediction.')
 flags.DEFINE_integer(
     'n_best_size', 20,
     'The total number of n-best predictions to generate in the '