Unverified Commit ca4e7aaa authored by Sang-Kil Park's avatar Sang-Kil Park Committed by GitHub
Browse files

Fix error when `bert_model` param is path or url.

Error occurs when `bert_model` param is path or url. Therefore, if it is path, specify the last path to prevent error.
parent 8da280eb
...@@ -855,7 +855,7 @@ def main(): ...@@ -855,7 +855,7 @@ def main():
global_step = 0 global_step = 0
if args.do_train: if args.do_train:
cached_train_features_file = args.train_file+'_{0}_{1}_{2}_{3}'.format( 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 train_features = None
try: try:
with open(cached_train_features_file, "rb") as reader: with open(cached_train_features_file, "rb") as reader:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment