Merge pull request #166 from likejazz/patch-1

Fix error when `bert_model` param is path or url.
2019-01-07 12:40:55 +01:00
parent 766c6b2ce3 ca4e7aaa72
commit d3d56f9a0b
1 changed files with 1 additions and 1 deletions
--- a/examples/run_squad.py
+++ b/examples/run_squad.py
@@ -856,7 +856,7 @@ def main():
    global_step = 0
    if args.do_train:
        cached_train_features_file = args.train_file+'_{0}_{1}_{2}_{3}'.format(
-            args.bert_model, str(args.max_seq_length), str(args.doc_stride), str(args.max_query_length))
+            list(filter(None, args.bert_model.split('/'))).pop(), str(args.max_seq_length), str(args.doc_stride), str(args.max_query_length))
        train_features = None
        try:
            with open(cached_train_features_file, "rb") as reader: