From f750586114d90275e1958cf39be267f5c9959ec5 Mon Sep 17 00:00:00 2001 From: Marcin Date: Tue, 26 Feb 2019 18:39:38 +0100 Subject: [PATCH] Remove redundant bptt param --- ulmfit/train_clas.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ulmfit/train_clas.py b/ulmfit/train_clas.py index f9a276e..f7a40ab 100644 --- a/ulmfit/train_clas.py +++ b/ulmfit/train_clas.py @@ -112,7 +112,7 @@ class CLSHyperParams(LMHyperParams): trn_args=dict(bptt=self.bptt, clip=self.clip) trn_args.update(kwargs) learn = text_classifier_learner(data_clas, AWD_LSTM, config=config, - pretrained=False, path=self.model_dir.parent, model_dir=self.model_dir.name, bptt=self.bptt, **trn_args) + pretrained=False, path=self.model_dir.parent, model_dir=self.model_dir.name, **trn_args) if self.pretrained_model is not None: print("Loading pretrained model")