small fix multi when common root in data_ids (OpenNMT#1444)

AMusketeer · May 23, 2019 · 7f1fc81 · 7f1fc81
1 parent 8ff0071
commit 7f1fc81
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onmt/inputters/inputter.py b/onmt/inputters/inputter.py
@@ -757,7 +757,7 @@ def build_dataset_iter(corpus_type, fields, opt, is_train=True, multi=False):
     but more sophisticated strategy like curriculum learning is ok too.
     """
     dataset_paths = list(sorted(
-        glob.glob(opt.data + '.' + corpus_type + '*.pt')))
+        glob.glob(opt.data + '.' + corpus_type + '.[0-9]*.pt')))
     if not dataset_paths:
         return None
     if multi: