search

google-research / timesfm

TimesFM (Time Series Foundation Model) is a pretrained time-series foundation model developed by Google Research for time-series forecasting.
https://research.google/blog/a-decoder-only-foundation-model-for-time-series-forecasting/
Apache License 2.0
3.64k stars 309 forks source link

## Checkpoint Loading Error in TimesFM #140

Open nmgvv opened 4 weeks ago

nmgvv commented 4 weeks ago

Description

When attempting to load a checkpoint for the TimesFM model, I encountered the following error:

The error suggests that for checkpoint versions > 1.0, a train_state_unpadded_shape_dtype_struct parameter is required during checkpoint loading. However, I couldn't find any documentation or examples in the TimesFM repository addressing this issue or demonstrating how to provide this parameter.

Environment

Steps to Reproduce

  1. [Provide steps to reproduce the error]
  2. [Include relevant code snippets if possible]

Questions

  1. Is this a known issue with recent versions of TimesFM or its dependencies?
  2. How can I provide the train_state_unpadded_shape_dtype_struct parameter when loading a checkpoint?
  3. Are there any plans to update the documentation or examples to address this new requirement?
  4. If this is related to a change in the underlying libraries (JAX/Flax), are there any compatibility notes or migration guides available?

Additional Context

I've checked the latest documentation and example code in the TimesFM repository but couldn't find any information about this new checkpoint loading requirement. Any guidance on resolving this issue or updating the checkpoint loading process would be greatly appreciated.

siriuz42 commented 3 weeks ago

It's shown as an error but should not block the model loading. Can you wait for another 2 minutes or so past the error is prompted?