mirror of
https://github.com/wassname/multifit.git
synced 2026-10-29 13:20:17 +08:00
Fix ftseed used in pretrain_lm so ulmfit lm works again
This commit is contained in:
1 parent
6ef75e4dd7
commit
b55715e4f6
1 file changed
+1
-1
@@ -191,7 +191,7 @@ class LMHyperParams:
|
||||
print("Saving info", self.model_dir / 'info.json')
|
||||
|
||||
def train_lm(self, num_epochs=20, data_lm=None, bs=70, true_wd=False, drop_mult=0.0, lr=5e-3, label_smoothing_eps=0.0):
|
||||
if self.ftseed is None:
|
||||
if not hasattr(self, 'ftseed'):
|
||||
self.set_seed(self.lmseed, "LM")
|
||||
else:
|
||||
self.set_seed(self.ftseed, "fine-tune")
|
||||
|
||||
Reference in new issue
Block a user