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

fix indent

Browse Source
This commit is contained in:
Aayush
2018-11-16 23:54:48 +05:30
committed by GitHub
parent 59c8852b5d
commit e72b19499c
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ulmfit/pretrain_lm.py
+1 -1
View File
@@ -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])