From bf33f222daaa2ae73a6aea08a1e4f614e34e1401 Mon Sep 17 00:00:00 2001 From: "NAUSICAA\\Julian" Date: Mon, 26 Nov 2018 22:48:42 -0300 Subject: [PATCH] callback > callback_fns --- 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 2396eaf..44dfa70 100644 --- a/ulmfit/pretrain_lm.py +++ b/ulmfit/pretrain_lm.py @@ -143,7 +143,7 @@ def pretrain_lm(dir_path, lang='en', cuda_id=0, qrnn=True, subword=False, max_vo learn = lm_learner(data_lm, bptt=bptt, emb_sz=emb_sz, nh=nh, nl=nl, pad_token=1, drop_mult=drop_mult, tie_weights=True, model_dir=model_dir.name, bias=True, qrnn=qrnn, clip=0.12, - callbacks=[lambda lrn: SaveModelCallback(lrn, every='epoch')]) + callback_fns=[lambda lrn: SaveModelCallback(lrn, every='epoch')]) # compared to standard Adam, we set beta_1 to 0.8 learn.opt_fn = partial(optim.Adam, betas=(0.8, 0.99))