diff --git a/examples/run_squad.py b/examples/run_squad.py
index 6a97dd300b..bbc8038676 100644
--- a/examples/run_squad.py
+++ b/examples/run_squad.py
@@ -855,7 +855,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: