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.02k stars 228 forks source link

the self._pmapped_decode interrupted, and does not output any errors #40

Closed mariomoonetedsfsd closed 1 month ago

mariomoonetedsfsd commented 1 month ago

checkpoint_path is : checkpoints Constructing model weights. Constructed model weights in 1.39 seconds. Restoring checkpoint from checkpoints. WARNING:absl:No registered CheckpointArgs found for handler type: <class 'paxml.checkpoints.FlaxCheckpointHandler'> WARNING:absl:Configured CheckpointManager using deprecated legacy API. Please follow the instructions at https://orbax.readthedocs.io/en/latest/api_refactor.html to migrate by May 1st, 2024. Restored checkpoint in 1.00 seconds. Jitting decoding

siriuz42 commented 1 month ago

Jitting will take a while (e.g. up to 1 minute on my server with 2x Intel Xeon Gold 6154). Any luck if you wait a bit longer?

mariomoonetedsfsd commented 1 month ago

checkpoint_path is : checkpoints Constructing model weights. Constructed model weights in 1.41 seconds. Restoring checkpoint from checkpoints. WARNING:absl:No registered CheckpointArgs found for handler type: <class 'paxml.checkpoints.FlaxCheckpointHandler'> WARNING:absl:Configured CheckpointManager using deprecated legacy API. Please follow the instructions at https://orbax.readthedocs.io/en/latest/api_refactor.html to migrate by May 1st, 2024. Restored checkpoint in 0.90 seconds. Jitting decoding.