Emaad commited on
Commit
f993da4
Β·
verified Β·
1 Parent(s): adbd260

Add model card: finetuning recipe + GIFT-Eval bar metrics + base-collection CTA

Browse files
Files changed (3) hide show
  1. .gitattributes +1 -0
  2. README.md +112 -0
  3. assets/bar_metrics_gift_eval.png +3 -0
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ assets/bar_metrics_gift_eval.png filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,112 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - time-series-forecasting
4
+ - foundation-models
5
+ - finetuned
6
+ - time-series
7
+ - timeseries
8
+ - forecasting
9
+ - observability
10
+ - gift-eval
11
+ - safetensors
12
+ - pytorch_model_hub_mixin
13
+ license: apache-2.0
14
+ pipeline_tag: time-series-forecasting
15
+ thumbnail: https://corp.dd-static.net/img/about/presskit/kit/press_kit.png
16
+ base_model: Datadog/Toto-2.0-2.5B
17
+ model-index:
18
+ - name: Toto-2.0-2.5B-FT
19
+ results:
20
+ - task:
21
+ type: time-series-forecasting
22
+ dataset:
23
+ name: GIFT-Eval
24
+ type: GIFT-Eval
25
+ metrics:
26
+ - name: CRPS
27
+ type: CRPS
28
+ value: 0.463
29
+ - name: MASE
30
+ type: MASE
31
+ value: 0.679
32
+ source:
33
+ name: GIFT-Eval Time Series Forecasting Leaderboard
34
+ url: https://huggingface.co/spaces/Salesforce/GIFT-Eval
35
+ ---
36
+
37
+ # Toto-2.0-2.5B-FT
38
+
39
+ > [!WARNING]
40
+ > **This is a benchmarking checkpoint, not a general-purpose model.**
41
+ > Toto-2.0-2.5B-FT is the [Toto 2.0 2.5B](https://huggingface.co/Datadog/Toto-2.0-2.5B) base model finetuned on the GIFT-Eval training split for our **#2-on-GIFT-Eval-leaderboard** submission. It is released for reproducibility only.
42
+ >
43
+ > For real workloads, please use the base [Toto 2.0 collection](https://huggingface.co/collections/Datadog/toto-20). The base checkpoints are pretrained without any public data, generalize to every benchmark we have evaluated, and are what we recommend deploying.
44
+
45
+ ---
46
+
47
+ ## ✨ What this is
48
+
49
+ A single Toto 2.0 2.5B base checkpoint finetuned on a mix that **includes the GIFT-Eval training split**, used to probe how far the base model can be pushed on a single in-distribution benchmark.
50
+
51
+ ![GIFT-Eval bar metrics β€” Toto 2.0 2.5B-FT highlighted](assets/bar_metrics_gift_eval.png)
52
+
53
+ On the full GIFT-Eval leaderboard (foundation models + finetuned + ensemble + agentic), Toto-2.0-2.5B-FT places **#2 on CRPS rank, MASE rank, and #3 on raw CRPS / MASE**, behind only the [Toto 2.0 Family-and-Friends](https://huggingface.co/Datadog/Toto-2.0-Family-and-Friends) ensemble.
54
+
55
+ ---
56
+
57
+ ## πŸ” Finetuning recipe
58
+
59
+ Starting from a fully-decayed [Toto-2.0-2.5B](https://huggingface.co/Datadog/Toto-2.0-2.5B) base checkpoint, we finetuned for 10,000 steps on a mix designed to expose the model to in-distribution structure without overfitting to GIFT-Eval alone:
60
+
61
+ | Source | Share |
62
+ |---|---|
63
+ | GIFT-Eval Pretrain | 45% |
64
+ | Datadog 5-minute+ observability metrics | 25% |
65
+ | GIFT-Eval train split | 15% |
66
+ | Synthetic (TempoPFN) | 10% |
67
+ | Datadog 10s observability metrics | 2.5% |
68
+ | Datadog 60s observability metrics | 2.5% |
69
+
70
+ The public portion (45% GIFT-Eval Pretrain) is drawn from the Toto 1.0 mix of GIFT-Eval Pretrain and the Chronos pretraining corpus, and is non-leaking with respect to the GIFT-Eval test split.
71
+
72
+ NorMuon and AdamW learning rates were both dropped by roughly an order of magnitude from pretraining (to 0.05 and 0.001 respectively). All other architecture and inference settings match the base 2.5B model.
73
+
74
+ ---
75
+
76
+ ## ⚑ Quick Start
77
+
78
+ ```python
79
+ import torch
80
+ from toto2 import Toto2Model
81
+
82
+ model = Toto2Model.from_pretrained("Datadog/Toto-2.0-2.5B-FT")
83
+ model = model.to("cuda").eval()
84
+
85
+ # Same forecast() interface as the base 2.5B model.
86
+ ```
87
+
88
+ See the base [Toto-2.0-2.5B](https://huggingface.co/Datadog/Toto-2.0-2.5B) model card for the full inference example.
89
+
90
+ ---
91
+
92
+ ## πŸ”— Additional Resources
93
+
94
+ - **Technical Report** β€” *(coming soon)*
95
+ - [Blog Post](https://www.datadoghq.com/blog/ai/toto-2/)
96
+ - [Base model: Toto-2.0-2.5B](https://huggingface.co/Datadog/Toto-2.0-2.5B) β€” the unfinetuned checkpoint, which is what we recommend deploying
97
+ - [Toto 2.0 Collection](https://huggingface.co/collections/Datadog/toto-20) β€” all five base sizes (4M β†’ 2.5B)
98
+ - [Toto 2.0 Family-and-Friends](https://huggingface.co/Datadog/Toto-2.0-Family-and-Friends) β€” companion FFORMA-ensemble submission, also benchmark-only
99
+ - [GIFT-Eval benchmark](https://huggingface.co/spaces/Salesforce/GIFT-Eval) β€” leaderboard hosting this submission
100
+ - [GitHub Repository](https://github.com/DataDog/toto)
101
+
102
+ ---
103
+
104
+ ## πŸ“ License
105
+
106
+ Apache 2.0.
107
+
108
+ ## πŸ“– Citation
109
+
110
+ ```bibtex
111
+ (citation coming soon)
112
+ ```
assets/bar_metrics_gift_eval.png ADDED

Git LFS Details

  • SHA256: 9eef1afe0c18126a9e4813345ae4b0189539c61878b4e0d3428e1205bfe13c5e
  • Pointer size: 131 Bytes
  • Size of remote file: 602 kB