Prediction changes for every run

amazon-science / chronos-forecasting

Chronos: Pretrained (Language) Models for Probabilistic Time Series Forecasting

https://arxiv.org/abs/2403.07815

Apache License 2.0

2.38k stars 273 forks source link

Prediction changes for every run #54

Closed harshitv804 closed 5 months ago

harshitv804 commented 5 months ago

Hi, I was trying to fine tuning the hyperparameters and every time I run the .predict, it gives me different predictions (sometimes converges very good, sometimes very bad). Any way to solve this Issue?

Thank you!

lostella commented 5 months ago

Hi @harshitv804, see the discussion in #18

lostella commented 5 months ago

The recommended way is

import transformers
transformers.set_seed(0)

harshitv804 commented 5 months ago

It's working. thanks !