Add hparams context manager

SalesforceAIResearch / uni2ts

[ICML2024] Unified Training of Universal Time Series Forecasting Transformers

Apache License 2.0

796 stars 80 forks source link

Closed gorold closed 5 months ago

gorold commented 5 months ago

This PR adds a hparams_context function to adjust the hparams of the MoiraiForecast object without reloading the model.

Credits to @abdulfatir for the original implementation.