From 84405b6e6c7e5e5f641fd68356ee4d99675ab933 Mon Sep 17 00:00:00 2001
From: Marcin <marcin.kardas@gmail.com>
Date: Mon, 13 May 2019 17:51:04 +0200
Subject: [PATCH] Allow bwd in model name

---
 ulmfit/__main__.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/ulmfit/__main__.py b/ulmfit/__main__.py
index 93e8b8a..be5ba92 100644
--- a/ulmfit/__main__.py
+++ b/ulmfit/__main__.py
@@ -34,9 +34,9 @@ def get_dataset_path(p, dataset_template):
     for ds_path in ds.parent.glob(pattern):
         yield lang, ds_path
 
-name_re = re.compile("(lstm|qrnn)_(.*)_(lmseed-)?.*\.m")
+name_re = re.compile("(bwd)?(lstm|qrnn)_(.*)_(lmseed-)?.*\.m")
 def folder_name_to_model_name(folder_name):
-    return name_re.match(folder_name).group(2)
+    return name_re.match(folder_name).group(3)
 
 class ULMFiT:
     @wraps(LMHyperParams)