Make MOMENT aware of insample mask, and freeze params

Nixtla / neuralforecast

Scalable and user friendly neural :brain: forecasting algorithms.

https://nixtlaverse.nixtla.io/neuralforecast

Apache License 2.0

2.69k stars 312 forks source link

Make MOMENT aware of insample mask, and freeze params #1014

Closed mononitogoswami closed 1 month ago

mononitogoswami commented 1 month ago

This commit: (1) add parameters to freeze MOMENT weights. By default all parameters except that of the forecasting head are frozen, and (2) makes MOMENT aware of insample_mask, such that it does not attend to padding time steps

review-notebook-app[bot] commented 1 month ago

Check out this pull request on

See visual diffs & provide feedback on Jupyter Notebooks.

Powered by ReviewNB

CLAassistant commented 1 month ago

All committers have signed the CLA.