search

amazon-science / chronos-forecasting

Chronos: Pretrained (Language) Models for Probabilistic Time Series Forecasting
https://arxiv.org/abs/2403.07815
Apache License 2.0
2.02k stars 238 forks source link

Enhance training script: auto tf32 detection and reorder default seed setting #91

Closed abdulfatir closed 4 weeks ago

abdulfatir commented 4 weeks ago

Description of changes: Automatically set tf32 to False if used on an older NVIDIA GPU. Reorder seed so that the seed is saved as part of the training config.

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.