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

Adding missing param arch

Browse Source
This commit is contained in:
NAUSICAA\Julian
2019-04-13 12:32:19 -03:00
parent 7744021014
commit edf3b8d76e
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_text_train.py
+2 -2
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, emb_sz=100, nl=1, drop_mult=0.1)
learn = language_model_learner(data, AWD_LSTM, emb_sz=100, nl=1, drop_mult=0.1)
learn.fit_one_cycle(4, 5e-3)
return learn
@@ -98,6 +98,6 @@ def test_bwdlm_lstm_can_be_trained():
data = TextLMDataBunch.from_df(path, df_trn, df_val, tokenizer=Tokenizer(BaseTokenizer),
lm_type = contrib_data.LanguageModelType.BwdLM)
learn = language_model_learner(data, emb_sz=100, nl=1, drop_mult=0.1, qrnn=False)
learn = language_model_learner(data, AWD_LSTM, emb_sz=100, nl=1, drop_mult=0.1, qrnn=False)
learn.fit_one_cycle(2, 5e-3)
assert learn.validate()[1] > 0.3