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 validate_cls in ulmfit module

Browse Source
This commit is contained in:
Piotr Czapla
2019-02-11 11:01:08 +01:00
parent c63fe258d2
commit a630242f97
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ulmfit/__main__.py
+1 -1
View File
@@ -20,7 +20,7 @@ class ULMFiT:
@wraps(CLSHyperParams)
def cls(self, dataset_path, base_lm_path, **changes):
params = CLSHyperParams.from_lm(dataset_path, base_lm_path, **changes)
return FireView(train=params.train_cls)
return FireView(train=params.train_cls, validate_cls=params.validate_cls)
if __name__ == '__main__':
fire.Fire(ULMFiT())