From 724233585e4a9229fde0e672770540aa5d78ff6e Mon Sep 17 00:00:00 2001 From: Piotr Czapla Date: Sun, 12 May 2019 22:19:02 +0200 Subject: [PATCH] Make it compatible with fastai 51 --- ulmfit/pretrain_lm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ulmfit/pretrain_lm.py b/ulmfit/pretrain_lm.py index b6f8458..1b55d04 100644 --- a/ulmfit/pretrain_lm.py +++ b/ulmfit/pretrain_lm.py @@ -305,7 +305,7 @@ class LMHyperParams: if (bunch_path / 'itos.pkl').exists(): data = bunch_class.load(self.cache_dir, name, bs=bs) elif bunch_path.exists(): - data = load_data(self.cache_dir, fname=name, bs=bs) + data = load_data(self.cache_dir, name, bs=bs) else: print(f"Running tokenization {name}...") data = bunch_class.from_df(path=self.cache_dir,