wassname/multifit
1
0
Fork 0
mirror of https://github.com/wassname/multifit.git synced 2026-09-09 11:27:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

Allow bwd in model name

Browse Source
This commit is contained in:
Marcin
2019-05-13 20:11:11 +02:00
parent b065e36f1b
commit 84405b6e6c
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ulmfit/__main__.py
+2 -2
View File
@@ -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)