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 resuming training of classifier

Browse Source
This commit is contained in:
Piotr Czapla
2018-11-17 17:13:15 +01:00
parent e97085337e
commit 2674a713fc
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ulmfit/train_clas.py
+1 -1
View File
@@ -130,7 +130,7 @@ def new_train_clas(data_dir, lang='en', cuda_id=0, pretrain_name='wt103', model_
drop_mult=0.3)
lm_enc_finetuned = f"{lm_name}_{dataset}_{name}_enc"
if fine_tune and not (model_dir / f"lm_enc_finetuned.pth").exists():
if fine_tune and not (model_dir/f"{lm_enc_finetuned}.pth").exists():
print('Fine-tuning the language model...')
learn.fit_one_cycle(1, 1e-2, moms=(0.8, 0.7))
learn.unfreeze()