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 issue in the databunch name generation

Browse Source
This commit is contained in:
Piotr Czapla
2019-05-18 21:48:32 +02:00
parent 8927e45fb2
commit f38f8c0670
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
@@ -321,7 +321,7 @@ class CLSHyperParams(LMHyperParams):
args['text_cols'] = list(trn_df.columns.values)[1:]
args['mark_fields'] = True
lm_suffix = self.bptt if self.bptt != 70 else ""
lm_suffix = self.use_tst_for_lm if "" else "-notst"
lm_suffix += self.use_tst_for_lm if "" else "-notst"
data_lm = self.lm_databunch(f'lm{lm_suffix}', train_df=lm_trn_df, valid_df=lm_val_df, bs=bs, force=force, bptt=self.bptt, **args)
args['vocab'] = data_lm.vocab
data_cls = self.cls_databunch(cls_name, train_df=trn_df, valid_df=val_df, bs=bs, force=force, **args)