From b55715e4f670ed8099ea109ff46f78cef2c934ee Mon Sep 17 00:00:00 2001 From: Piotr Czapla Date: Mon, 13 May 2019 17:47:28 +0200 Subject: [PATCH] Fix ftseed used in pretrain_lm so ulmfit lm works again --- 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 c3da68f..03bf4d2 100644 --- a/ulmfit/pretrain_lm.py +++ b/ulmfit/pretrain_lm.py @@ -191,7 +191,7 @@ class LMHyperParams: print("Saving info", self.model_dir / 'info.json') def train_lm(self, num_epochs=20, data_lm=None, bs=70, true_wd=False, drop_mult=0.0, lr=5e-3, label_smoothing_eps=0.0): - if self.ftseed is None: + if not hasattr(self, 'ftseed'): self.set_seed(self.lmseed, "LM") else: self.set_seed(self.ftseed, "fine-tune")