Elgato_dark/chronos-forecasting
1
0
Fork 0
mirror of https://github.com/amazon-science/chronos-forecasting synced 2026-05-24 01:58:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Chronos-2: Do not catch exceptions in predict_fev when fine-tuning (#417)

Browse source 
*Issue #, if available:*

*Description of changes:*


By submitting this pull request, I confirm that you can use, modify,
copy, and redistribute this contribution, under the terms of your
choice.
This commit is contained in:
Abdul Fatir 2025-12-03 15:33:08 +01:00 committed by GitHub
parent 9afe64332f
commit 7513dee92d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
src/chronos/chronos2/pipeline.py
View file

@ -1030,11 +1030,7 @@ class Chronos2Pipeline(BaseChronosPipeline):
finetune_kwargs["prediction_length"] = first_window.horizon
finetune_kwargs["batch_size"] = finetune_kwargs.get("batch_size", batch_size)
try:
pipeline = self.fit(inputs=inputs, **finetune_kwargs)
except Exception as e:
msg = f"Finetuning failed with error: {e}. Continuing with the pretrained model."
warnings.warn(msg, category=UserWarning, stacklevel=2)
pipeline = self.fit(inputs=inputs, **finetune_kwargs)
predictions_per_window = []
inference_time_s = 0.0