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

Expose num_lm_epochs in ulmfit eval

Browse Source
This commit is contained in:
Piotr Czapla
2019-02-15 01:11:39 +01:00
parent 0f084168c1
commit 0e6534ad7b
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ulmfit/__main__.py
+2 -2
View File
@@ -41,7 +41,7 @@ class ULMFiT:
params = CLSHyperParams.from_lm(dataset_path, base_lm_path, **changes)
return FireView(train=params.train_cls, validate_cls=params.validate_cls)
def eval(self, glob="mldoc/*-1/models/sp30k/lstm_nl4.m", dataset_template='${lang}-1', name="tmp-100", cuda_id=0, **trn_params):
def eval(self, glob="mldoc/*-1/models/sp30k/lstm_nl4.m", dataset_template='${lang}-1', name="tmp-100", num_lm_epochs=0, cuda_id=0, **trn_params):
results = OrderedDict()
for base_model in sorted(Path("data").glob(glob)):
for lang, dataset_path in sorted(get_dataset_path(base_model, dataset_template)):
@@ -52,7 +52,7 @@ class ULMFiT:
results[key] = params.validate_cls()[1]
else:
print("Training")
results[key] = params.train_cls(num_lm_epochs=0, **trn_params)[1]
results[key] = params.train_cls(num_lm_epochs=num_lm_epochs, **trn_params)[1]
del params
gc.collect()