chronos-forecasting/scripts/training
Abdul Fatir 6bcd4584a3
Enhance training script: auto tf32 detection and reorder default seed setting (#91)
*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.

---------

Co-authored-by: Abdul Fatir Ansari <ansarnd@amazon.de>
2024-05-31 15:13:49 +02:00
..
configs Add training script (#63) 2024-05-09 17:52:01 +02:00
train.py Enhance training script: auto tf32 detection and reorder default seed setting (#91) 2024-05-31 15:13:49 +02:00