run_squad WIP

2018-11-02 03:56:14 +01:00
parent c0065af6cb
commit e61db0d1c0
3 changed files with 75 additions and 40 deletions
--- a/extract_features_pytorch.py
+++ b/extract_features_pytorch.py
@@ -316,7 +316,7 @@ def read_examples(input_file):
    return examples


-def main(_):
+def main():
    tf.logging.set_verbosity(tf.logging.INFO)

    layer_indexes = [int(x) for x in args.layers.split(",")]
@@ -387,4 +387,4 @@ def main(_):


 if __name__ == "__main__":
-    tf.app.run()
+    main()