kevinduh
diff --git a/‎my_utils/__init__.py
100644100755 b/‎my_utils/__init__.py
100644100755
diff --git a/‎my_utils/data_utils.py
100644100755 b/‎my_utils/data_utils.py
100644100755
diff --git a/‎my_utils/log_wrapper.py
100644100755 b/‎my_utils/log_wrapper.py
100644100755
diff --git a/‎my_utils/squad_eval.py
100644100755 b/‎my_utils/squad_eval.py
100644100755
diff --git a/‎my_utils/squad_eval_v2.py
100644100755 b/‎my_utils/squad_eval_v2.py
100644100755
diff --git a/‎my_utils/tokenizer.py
100644100755 b/‎my_utils/tokenizer.py
100644100755
diff --git a/‎my_utils/utils.py
100644100755 b/‎my_utils/utils.py
100644100755
diff --git a/‎my_utils/word2vec_utils.py
100644100755 b/‎my_utils/word2vec_utils.py
100644100755
diff --git a/‎train.py
Lines changed: 1 addition & 1 deletion b/‎train.py
Lines changed: 1 addition & 1 deletion
@@ -73,7 +73,7 @@ def main():
     if os.path.exists(test_path):
         test_data = BatchGen(test_path,
                             batch_size=args.batch_size,
-                            gpu=args.cuda, is_train=False)
+                            gpu=args.cuda, is_train=False, elmo_on=args.elmo_on)
 
     # load golden standard
     dev_gold = load_squad(dev_gold_path)