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

Remove config params from language_model_learner funcion

Browse Source
This commit is contained in:
NAUSICAA\Julian
2019-04-13 16:08:01 -03:00
parent edf3b8d76e
commit 7e0a8517b6
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_text_train.py
+1 -1
View File
@@ -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