From e72b19499c0c7a909d7a139259dbb85a97f2c0d8 Mon Sep 17 00:00:00 2001 From: Aayush Date: Fri, 16 Nov 2018 23:54:48 +0530 Subject: [PATCH] fix indent --- 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 12ce63f..4a16530 100644 --- a/ulmfit/pretrain_lm.py +++ b/ulmfit/pretrain_lm.py @@ -91,7 +91,7 @@ def pretrain_lm(dir_path, lang='en', cuda_id=0, qrnn=True, subword=False, max_vo # save vocabulary print(f"Saving vocabulary as {dir_path / model_dir}") with open(dir_path / model_dir / f'itos_{name}.pkl', 'wb') as f: - pickle.dump(itos, f) + pickle.dump(itos, f) trn_ids = np.array([([stoi.get(w, stoi[UNK]) for w in s]) for s in trn_tok])