From 7e0a8517b6f577635bf1a0aab7774ae439a0fa04 Mon Sep 17 00:00:00 2001 From: "NAUSICAA\\Julian" Date: Sat, 13 Apr 2019 16:08:01 -0300 Subject: [PATCH] Remove config params from language_model_learner funcion --- tests/test_text_train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_text_train.py b/tests/test_text_train.py index 0f41de5..ccc9bef 100644 --- a/tests/test_text_train.py +++ b/tests/test_text_train.py @@ -36,7 +36,7 @@ def manual_seed(seed=42): def learn(): path, df_trn, df_val = prep_human_numbers() data = TextLMDataBunch.from_df(path, df_trn, df_val, tokenizer=Tokenizer(BaseTokenizer)) - learn = language_model_learner(data, AWD_LSTM, emb_sz=100, nl=1, drop_mult=0.1) + learn = language_model_learner(data, AWD_LSTM) learn.fit_one_cycle(4, 5e-3) return learn