mirror of
https://github.com/wassname/multifit.git
synced 2026-10-30 13:30:16 +08:00
19 lines
432 B
Python
19 lines
432 B
Python
from pathlib import Path
|
|
|
|
import fire
|
|
|
|
import ulmfit.configurations
|
|
import ulmfit
|
|
|
|
class Experiment:
|
|
def new(self):
|
|
return {n: getattr(ulmfit.configurations,n) for n in ulmfit.configurations.__all__}
|
|
|
|
def load(self, model_path):
|
|
return ulmfit.ULMFiT().load_(Path(model_path))
|
|
|
|
def from_pretrained(self):
|
|
return ulmfit.ULMFiT.from_pretrained
|
|
|
|
if __name__ == '__main__':
|
|
fire.Fire(Experiment()) |