search

moment-timeseries-foundation-model / moment

MOMENT: A Family of Open Time-series Foundation Models
https://moment-timeseries-foundation-model.github.io/
MIT License
365 stars 59 forks source link

Can't load up model with classification class #55

Closed royweiss1 closed 2 months ago

royweiss1 commented 2 months ago

Hi, when I try to use the model with: model = MOMENTPipeline.from_pretrained( "AutonLab/MOMENT-1-large", model_kwargs={ 'task_name': 'classification', 'n_channels': 1, 'num_class': num_classes, }, )

And then I print: print(model.task_name) I get: reconstruction

Please help me figure out why. Thanks

royweiss1 commented 2 months ago

lol nevermind I forgot to write: model.init()