GritLs commited on
Commit
b30c45e
·
verified ·
1 Parent(s): 5b81771

Add files using upload-large-folder tool

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. experiments/bridge/synth-m/seed0/config.template.yaml +169 -0
  2. experiments/bridge/synth-m/seed0/results/expected_seed_metrics.json +21 -0
  3. experiments/bridge/synth-m/seed0/summary.json +5 -0
  4. experiments/bridge/synth-m/seed1/config.template.yaml +169 -0
  5. experiments/bridge/synth-m/seed1/results/expected_seed_metrics.json +21 -0
  6. experiments/bridge/synth-m/seed1/summary.json +5 -0
  7. experiments/bridge/synth-m/seed2/config.template.yaml +169 -0
  8. experiments/bridge/synth-m/seed2/results/expected_seed_metrics.json +21 -0
  9. experiments/bridge/synth-m/seed2/summary.json +5 -0
  10. experiments/bridge/synth-u/seed0/config.template.yaml +169 -0
  11. experiments/bridge/synth-u/seed0/results/expected_seed_metrics.json +21 -0
  12. experiments/bridge/synth-u/seed0/summary.json +5 -0
  13. experiments/bridge/synth-u/seed1/config.template.yaml +169 -0
  14. experiments/bridge/synth-u/seed1/results/expected_seed_metrics.json +21 -0
  15. experiments/bridge/synth-u/seed1/summary.json +5 -0
  16. experiments/bridge/synth-u/seed2/config.template.yaml +169 -0
  17. experiments/bridge/synth-u/seed2/results/expected_seed_metrics.json +21 -0
  18. experiments/bridge/synth-u/seed2/summary.json +5 -0
  19. experiments/diffusets/synth-m/seed0/config.template.yaml +156 -0
  20. experiments/diffusets/synth-m/seed0/results/expected_seed_metrics.json +21 -0
  21. experiments/diffusets/synth-m/seed0/summary.json +5 -0
  22. experiments/diffusets/synth-m/seed1/config.template.yaml +156 -0
  23. experiments/diffusets/synth-m/seed1/results/expected_seed_metrics.json +21 -0
  24. experiments/diffusets/synth-m/seed1/summary.json +5 -0
  25. experiments/diffusets/synth-m/seed2/config.template.yaml +156 -0
  26. experiments/diffusets/synth-m/seed2/results/expected_seed_metrics.json +21 -0
  27. experiments/diffusets/synth-m/seed2/summary.json +5 -0
  28. experiments/diffusets/synth-u/seed0/config.template.yaml +156 -0
  29. experiments/diffusets/synth-u/seed0/results/expected_seed_metrics.json +21 -0
  30. experiments/diffusets/synth-u/seed0/summary.json +5 -0
  31. experiments/diffusets/synth-u/seed1/config.template.yaml +156 -0
  32. experiments/diffusets/synth-u/seed1/results/expected_seed_metrics.json +21 -0
  33. experiments/diffusets/synth-u/seed1/summary.json +5 -0
  34. experiments/diffusets/synth-u/seed2/config.template.yaml +156 -0
  35. experiments/diffusets/synth-u/seed2/results/expected_seed_metrics.json +21 -0
  36. experiments/diffusets/synth-u/seed2/summary.json +5 -0
  37. experiments/t2s/synth-m/seed0/config.template.yaml +153 -0
  38. experiments/t2s/synth-m/seed0/results/expected_seed_metrics.json +21 -0
  39. experiments/t2s/synth-m/seed0/summary.json +5 -0
  40. experiments/t2s/synth-m/seed1/config.template.yaml +153 -0
  41. experiments/t2s/synth-m/seed1/results/expected_seed_metrics.json +21 -0
  42. experiments/t2s/synth-m/seed1/summary.json +5 -0
  43. experiments/t2s/synth-m/seed2/config.template.yaml +153 -0
  44. experiments/t2s/synth-m/seed2/results/expected_seed_metrics.json +21 -0
  45. experiments/t2s/synth-m/seed2/summary.json +5 -0
  46. experiments/t2s/synth-u/seed0/config.template.yaml +153 -0
  47. experiments/t2s/synth-u/seed0/results/expected_seed_metrics.json +21 -0
  48. experiments/t2s/synth-u/seed0/summary.json +5 -0
  49. experiments/t2s/synth-u/seed1/config.template.yaml +153 -0
  50. experiments/t2s/synth-u/seed1/results/expected_seed_metrics.json +21 -0
experiments/bridge/synth-m/seed0/config.template.yaml ADDED
@@ -0,0 +1,169 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: synth-m_bridge
2
+ description: ''
3
+ seed: 0
4
+ device: cuda:0
5
+ train:
6
+ stages:
7
+ - name: finetune
8
+ epochs: 700
9
+ lr: 0.001
10
+ use_condition: true
11
+ freeze_modules: []
12
+ early_stopping_patience: 50
13
+ epochs: 700
14
+ batch_size: 256
15
+ lr: 0.001
16
+ weight_decay: 0.0001
17
+ scheduler: cosine
18
+ scheduler_params: {}
19
+ early_stopping_patience: 50
20
+ gradient_clip_val: 1.0
21
+ accumulate_grad_batches: 1
22
+ val_check_interval: 1.0
23
+ num_workers: 4
24
+ pin_memory: true
25
+ limit_train_batches: 1.0
26
+ limit_val_batches: 1.0
27
+ limit_test_batches: 1.0
28
+ num_sanity_val_steps: 2
29
+ log_grad_norm: true
30
+ log_param_norm: true
31
+ log_norm_every_n_steps: 50
32
+ data:
33
+ name: synth-m
34
+ data_folder: __DATA_ROOT__/datasets/synth-m
35
+ n_var: 2
36
+ seq_length: 128
37
+ normalize: false
38
+ train_split: 0.8
39
+ val_split: 0.1
40
+ model:
41
+ name: bridge
42
+ channels: 64
43
+ layers: 3
44
+ nheads: 8
45
+ dropout: 0.0
46
+ latent_dim: 32
47
+ num_latents: 16
48
+ latent_unit: 1
49
+ repre_emb_channels: 32
50
+ prototype_dim: 32
51
+ context_dim: 32
52
+ text_dim: 1024
53
+ fusion_type: gated_add
54
+ cond_drop_prob: 0.5
55
+ use_cfg: true
56
+ model_channels: 64
57
+ num_res_blocks: 2
58
+ attention_resolutions:
59
+ - 1
60
+ - 2
61
+ - 4
62
+ channel_mult:
63
+ - 1
64
+ - 2
65
+ - 4
66
+ - 4
67
+ num_heads: 8
68
+ num_head_channels: -1
69
+ use_scale_shift_norm: true
70
+ resblock_updown: true
71
+ use_spatial_transformer: true
72
+ transformer_depth: 1
73
+ diffusion_steps: 50
74
+ noise_schedule: quad
75
+ beta_start: 0.0001
76
+ beta_end: 0.5
77
+ bridge_type: vp
78
+ condition:
79
+ text:
80
+ enabled: true
81
+ input_dim: 1024
82
+ embedding_key: cap_emb
83
+ dropout: 0.0
84
+ cfg_scale: 1.0
85
+ attribute:
86
+ enabled: false
87
+ continuous_dim: 0
88
+ discrete_configs: []
89
+ output_dim: 128
90
+ dropout: 0.0
91
+ label:
92
+ enabled: false
93
+ num_classes: 10
94
+ output_dim: 64
95
+ fusion: concat
96
+ condition_dropout: 0.0
97
+ eval:
98
+ n_samples: 10
99
+ metrics:
100
+ - dtw
101
+ - wape
102
+ - ed
103
+ - crps
104
+ - acd
105
+ - sd
106
+ - kd
107
+ - mdd
108
+ - fid
109
+ - sfd
110
+ - jftsd
111
+ - prdc_f1
112
+ - joint_prdc_f1
113
+ - cttp
114
+ - disc_auc
115
+ batch_size: 128
116
+ save_samples: true
117
+ sampler: ddim
118
+ display_interval: 10
119
+ clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-m/model_configs.yaml
120
+ clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-m/clip_model_best.pth
121
+ cache_folder: __CACHE_ROOT__/bridge/synth-m/seed0
122
+ use_longalign: false
123
+ reference_split: train
124
+ disc_auc:
125
+ k_folds: 5
126
+ epochs: 5
127
+ batch_size: 128
128
+ lr: 0.001
129
+ weight_decay: 0.0001
130
+ hidden_channels: 128
131
+ num_layers: 3
132
+ dropout: 0.1
133
+ seed: 42
134
+ num_workers: 0
135
+ prdc:
136
+ k: 5
137
+ max_samples: 5000
138
+ seed: 0
139
+ metric: euclidean
140
+ backend: auto
141
+ knn_chunk_size: 4096
142
+ joint_enable: true
143
+ joint_weights:
144
+ ts: 1.0
145
+ text: 1.0
146
+ joint_normalize: standard
147
+ segment_classifier:
148
+ enable: false
149
+ segment_len: 128
150
+ n_segments: 3
151
+ seed: 42
152
+ acd_max_lag: 50
153
+ mdd_bins: 32
154
+ use_cache: false
155
+ cache_file: predictions_cache.pkl
156
+ viz:
157
+ enable: true
158
+ k_cases: 20
159
+ max_vars: 8
160
+ ncols: 5
161
+ seed: 42
162
+ dpi: 200
163
+ alpha: 0.25
164
+ figsize_per_subplot:
165
+ - 3.0
166
+ - 2.0
167
+ output_type: pdf
168
+ output_dir: ''
169
+ created_at: '2025-12-28T18:30:19.791690'
experiments/bridge/synth-m/seed0/results/expected_seed_metrics.json ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dtw": 13.817281705569064,
3
+ "wape": 106.19159049917226,
4
+ "ed": 16.047494094848634,
5
+ "crps": 0.6222953609760046,
6
+ "acd": 0.04810057580471039,
7
+ "sd": 0.3342529663983614,
8
+ "kd": 0.2772295471154491,
9
+ "mdd": 0.029095235188802075,
10
+ "fid": 36.90573498644184,
11
+ "sfd": 46.77665048709039,
12
+ "jftsd": 44.922412953420775,
13
+ "prdc_f1.f1": 0.2880602782071098,
14
+ "prdc_f1.precision": 0.62125,
15
+ "prdc_f1.recall": 0.1875,
16
+ "joint_prdc_f1.f1": 0.33039791744142805,
17
+ "joint_prdc_f1.precision": 0.38,
18
+ "joint_prdc_f1.recall": 0.29225,
19
+ "cttp": 18.433125,
20
+ "disc_auc": 0.9997595312499999
21
+ }
experiments/bridge/synth-m/seed0/summary.json ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ {
2
+ "status": "completed",
3
+ "finished_at": "2025-12-28T22:05:24.380146",
4
+ "best_checkpoint": "checkpoints/finetune/best.ckpt"
5
+ }
experiments/bridge/synth-m/seed1/config.template.yaml ADDED
@@ -0,0 +1,169 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: synth-m_bridge
2
+ description: ''
3
+ seed: 1
4
+ device: cuda:0
5
+ train:
6
+ stages:
7
+ - name: finetune
8
+ epochs: 700
9
+ lr: 0.001
10
+ use_condition: true
11
+ freeze_modules: []
12
+ early_stopping_patience: 50
13
+ epochs: 700
14
+ batch_size: 256
15
+ lr: 0.001
16
+ weight_decay: 0.0001
17
+ scheduler: cosine
18
+ scheduler_params: {}
19
+ early_stopping_patience: 50
20
+ gradient_clip_val: 1.0
21
+ accumulate_grad_batches: 1
22
+ val_check_interval: 1.0
23
+ num_workers: 4
24
+ pin_memory: true
25
+ limit_train_batches: 1.0
26
+ limit_val_batches: 1.0
27
+ limit_test_batches: 1.0
28
+ num_sanity_val_steps: 2
29
+ log_grad_norm: true
30
+ log_param_norm: true
31
+ log_norm_every_n_steps: 50
32
+ data:
33
+ name: synth-m
34
+ data_folder: __DATA_ROOT__/datasets/synth-m
35
+ n_var: 2
36
+ seq_length: 128
37
+ normalize: false
38
+ train_split: 0.8
39
+ val_split: 0.1
40
+ model:
41
+ name: bridge
42
+ channels: 64
43
+ layers: 3
44
+ nheads: 8
45
+ dropout: 0.0
46
+ latent_dim: 32
47
+ num_latents: 16
48
+ latent_unit: 1
49
+ repre_emb_channels: 32
50
+ prototype_dim: 32
51
+ context_dim: 32
52
+ text_dim: 1024
53
+ fusion_type: gated_add
54
+ cond_drop_prob: 0.5
55
+ use_cfg: true
56
+ model_channels: 64
57
+ num_res_blocks: 2
58
+ attention_resolutions:
59
+ - 1
60
+ - 2
61
+ - 4
62
+ channel_mult:
63
+ - 1
64
+ - 2
65
+ - 4
66
+ - 4
67
+ num_heads: 8
68
+ num_head_channels: -1
69
+ use_scale_shift_norm: true
70
+ resblock_updown: true
71
+ use_spatial_transformer: true
72
+ transformer_depth: 1
73
+ diffusion_steps: 50
74
+ noise_schedule: quad
75
+ beta_start: 0.0001
76
+ beta_end: 0.5
77
+ bridge_type: vp
78
+ condition:
79
+ text:
80
+ enabled: true
81
+ input_dim: 1024
82
+ embedding_key: cap_emb
83
+ dropout: 0.0
84
+ cfg_scale: 1.0
85
+ attribute:
86
+ enabled: false
87
+ continuous_dim: 0
88
+ discrete_configs: []
89
+ output_dim: 128
90
+ dropout: 0.0
91
+ label:
92
+ enabled: false
93
+ num_classes: 10
94
+ output_dim: 64
95
+ fusion: concat
96
+ condition_dropout: 0.0
97
+ eval:
98
+ n_samples: 10
99
+ metrics:
100
+ - dtw
101
+ - wape
102
+ - ed
103
+ - crps
104
+ - acd
105
+ - sd
106
+ - kd
107
+ - mdd
108
+ - fid
109
+ - sfd
110
+ - jftsd
111
+ - prdc_f1
112
+ - joint_prdc_f1
113
+ - cttp
114
+ - disc_auc
115
+ batch_size: 128
116
+ save_samples: true
117
+ sampler: ddim
118
+ display_interval: 10
119
+ clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-m/model_configs.yaml
120
+ clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-m/clip_model_best.pth
121
+ cache_folder: __CACHE_ROOT__/bridge/synth-m/seed1
122
+ use_longalign: false
123
+ reference_split: train
124
+ disc_auc:
125
+ k_folds: 5
126
+ epochs: 5
127
+ batch_size: 128
128
+ lr: 0.001
129
+ weight_decay: 0.0001
130
+ hidden_channels: 128
131
+ num_layers: 3
132
+ dropout: 0.1
133
+ seed: 42
134
+ num_workers: 0
135
+ prdc:
136
+ k: 5
137
+ max_samples: 5000
138
+ seed: 0
139
+ metric: euclidean
140
+ backend: auto
141
+ knn_chunk_size: 4096
142
+ joint_enable: true
143
+ joint_weights:
144
+ ts: 1.0
145
+ text: 1.0
146
+ joint_normalize: standard
147
+ segment_classifier:
148
+ enable: false
149
+ segment_len: 128
150
+ n_segments: 3
151
+ seed: 42
152
+ acd_max_lag: 50
153
+ mdd_bins: 32
154
+ use_cache: false
155
+ cache_file: predictions_cache.pkl
156
+ viz:
157
+ enable: true
158
+ k_cases: 20
159
+ max_vars: 8
160
+ ncols: 5
161
+ seed: 42
162
+ dpi: 200
163
+ alpha: 0.25
164
+ figsize_per_subplot:
165
+ - 3.0
166
+ - 2.0
167
+ output_type: pdf
168
+ output_dir: ''
169
+ created_at: '2025-12-28T18:32:05.306146'
experiments/bridge/synth-m/seed1/results/expected_seed_metrics.json ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dtw": 11.941985574008141,
3
+ "wape": 97.19576751179419,
4
+ "ed": 14.614929718017578,
5
+ "crps": 0.5933726656309234,
6
+ "acd": 0.08524112403392792,
7
+ "sd": 0.027004171864705717,
8
+ "kd": 0.707300600732306,
9
+ "mdd": 0.02950222778320312,
10
+ "fid": 60.61234598914979,
11
+ "sfd": 45.196805251014396,
12
+ "jftsd": 68.15098610695699,
13
+ "prdc_f1.f1": 0.17119105807935597,
14
+ "prdc_f1.precision": 0.77325,
15
+ "prdc_f1.recall": 0.09625,
16
+ "joint_prdc_f1.f1": 0.19173251468232783,
17
+ "joint_prdc_f1.precision": 0.33375,
18
+ "joint_prdc_f1.recall": 0.1345,
19
+ "cttp": 18.584201385498048,
20
+ "disc_auc": 0.9996928125
21
+ }
experiments/bridge/synth-m/seed1/summary.json ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ {
2
+ "status": "completed",
3
+ "finished_at": "2025-12-28T19:25:34.577642",
4
+ "best_checkpoint": "checkpoints/finetune/best.ckpt"
5
+ }
experiments/bridge/synth-m/seed2/config.template.yaml ADDED
@@ -0,0 +1,169 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: synth-m_bridge
2
+ description: ''
3
+ seed: 2
4
+ device: cuda:0
5
+ train:
6
+ stages:
7
+ - name: finetune
8
+ epochs: 700
9
+ lr: 0.001
10
+ use_condition: true
11
+ freeze_modules: []
12
+ early_stopping_patience: 50
13
+ epochs: 700
14
+ batch_size: 256
15
+ lr: 0.001
16
+ weight_decay: 0.0001
17
+ scheduler: cosine
18
+ scheduler_params: {}
19
+ early_stopping_patience: 50
20
+ gradient_clip_val: 1.0
21
+ accumulate_grad_batches: 1
22
+ val_check_interval: 1.0
23
+ num_workers: 4
24
+ pin_memory: true
25
+ limit_train_batches: 1.0
26
+ limit_val_batches: 1.0
27
+ limit_test_batches: 1.0
28
+ num_sanity_val_steps: 2
29
+ log_grad_norm: true
30
+ log_param_norm: true
31
+ log_norm_every_n_steps: 50
32
+ data:
33
+ name: synth-m
34
+ data_folder: __DATA_ROOT__/datasets/synth-m
35
+ n_var: 2
36
+ seq_length: 128
37
+ normalize: false
38
+ train_split: 0.8
39
+ val_split: 0.1
40
+ model:
41
+ name: bridge
42
+ channels: 64
43
+ layers: 3
44
+ nheads: 8
45
+ dropout: 0.0
46
+ latent_dim: 32
47
+ num_latents: 16
48
+ latent_unit: 1
49
+ repre_emb_channels: 32
50
+ prototype_dim: 32
51
+ context_dim: 32
52
+ text_dim: 1024
53
+ fusion_type: gated_add
54
+ cond_drop_prob: 0.5
55
+ use_cfg: true
56
+ model_channels: 64
57
+ num_res_blocks: 2
58
+ attention_resolutions:
59
+ - 1
60
+ - 2
61
+ - 4
62
+ channel_mult:
63
+ - 1
64
+ - 2
65
+ - 4
66
+ - 4
67
+ num_heads: 8
68
+ num_head_channels: -1
69
+ use_scale_shift_norm: true
70
+ resblock_updown: true
71
+ use_spatial_transformer: true
72
+ transformer_depth: 1
73
+ diffusion_steps: 50
74
+ noise_schedule: quad
75
+ beta_start: 0.0001
76
+ beta_end: 0.5
77
+ bridge_type: vp
78
+ condition:
79
+ text:
80
+ enabled: true
81
+ input_dim: 1024
82
+ embedding_key: cap_emb
83
+ dropout: 0.0
84
+ cfg_scale: 1.0
85
+ attribute:
86
+ enabled: false
87
+ continuous_dim: 0
88
+ discrete_configs: []
89
+ output_dim: 128
90
+ dropout: 0.0
91
+ label:
92
+ enabled: false
93
+ num_classes: 10
94
+ output_dim: 64
95
+ fusion: concat
96
+ condition_dropout: 0.0
97
+ eval:
98
+ n_samples: 10
99
+ metrics:
100
+ - dtw
101
+ - wape
102
+ - ed
103
+ - crps
104
+ - acd
105
+ - sd
106
+ - kd
107
+ - mdd
108
+ - fid
109
+ - sfd
110
+ - jftsd
111
+ - prdc_f1
112
+ - joint_prdc_f1
113
+ - cttp
114
+ - disc_auc
115
+ batch_size: 128
116
+ save_samples: true
117
+ sampler: ddim
118
+ display_interval: 10
119
+ clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-m/model_configs.yaml
120
+ clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-m/clip_model_best.pth
121
+ cache_folder: __CACHE_ROOT__/bridge/synth-m/seed2
122
+ use_longalign: false
123
+ reference_split: train
124
+ disc_auc:
125
+ k_folds: 5
126
+ epochs: 5
127
+ batch_size: 128
128
+ lr: 0.001
129
+ weight_decay: 0.0001
130
+ hidden_channels: 128
131
+ num_layers: 3
132
+ dropout: 0.1
133
+ seed: 42
134
+ num_workers: 0
135
+ prdc:
136
+ k: 5
137
+ max_samples: 5000
138
+ seed: 0
139
+ metric: euclidean
140
+ backend: auto
141
+ knn_chunk_size: 4096
142
+ joint_enable: true
143
+ joint_weights:
144
+ ts: 1.0
145
+ text: 1.0
146
+ joint_normalize: standard
147
+ segment_classifier:
148
+ enable: false
149
+ segment_len: 128
150
+ n_segments: 3
151
+ seed: 42
152
+ acd_max_lag: 50
153
+ mdd_bins: 32
154
+ use_cache: false
155
+ cache_file: predictions_cache.pkl
156
+ viz:
157
+ enable: true
158
+ k_cases: 20
159
+ max_vars: 8
160
+ ncols: 5
161
+ seed: 42
162
+ dpi: 200
163
+ alpha: 0.25
164
+ figsize_per_subplot:
165
+ - 3.0
166
+ - 2.0
167
+ output_type: pdf
168
+ output_dir: ''
169
+ created_at: '2025-12-28T18:35:18.501692'
experiments/bridge/synth-m/seed2/results/expected_seed_metrics.json ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dtw": 12.013450861567545,
3
+ "wape": 96.39575187473649,
4
+ "ed": 14.510944213867187,
5
+ "crps": 0.5734683085540916,
6
+ "acd": 0.04058488830924034,
7
+ "sd": 0.05335535791835336,
8
+ "kd": 0.2815472187496497,
9
+ "mdd": 0.029644704182942704,
10
+ "fid": 40.9487228029339,
11
+ "sfd": 42.799137914105,
12
+ "jftsd": 47.69404022174564,
13
+ "prdc_f1.f1": 0.2721332174920359,
14
+ "prdc_f1.precision": 0.694,
15
+ "prdc_f1.recall": 0.16925,
16
+ "joint_prdc_f1.f1": 0.35218493819128177,
17
+ "joint_prdc_f1.precision": 0.49525,
18
+ "joint_prdc_f1.recall": 0.27325,
19
+ "cttp": 19.501730834960938,
20
+ "disc_auc": 0.99973171875
21
+ }
experiments/bridge/synth-m/seed2/summary.json ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ {
2
+ "status": "completed",
3
+ "finished_at": "2025-12-28T20:09:44.682046",
4
+ "best_checkpoint": "checkpoints/finetune/best.ckpt"
5
+ }
experiments/bridge/synth-u/seed0/config.template.yaml ADDED
@@ -0,0 +1,169 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: synth-u_bridge
2
+ description: ''
3
+ seed: 0
4
+ device: cuda:0
5
+ train:
6
+ stages:
7
+ - name: finetune
8
+ epochs: 700
9
+ lr: 0.001
10
+ use_condition: true
11
+ freeze_modules: []
12
+ early_stopping_patience: 50
13
+ epochs: 700
14
+ batch_size: 256
15
+ lr: 0.001
16
+ weight_decay: 0.0001
17
+ scheduler: cosine
18
+ scheduler_params: {}
19
+ early_stopping_patience: 50
20
+ gradient_clip_val: 1.0
21
+ accumulate_grad_batches: 1
22
+ val_check_interval: 1.0
23
+ num_workers: 4
24
+ pin_memory: true
25
+ limit_train_batches: 1.0
26
+ limit_val_batches: 1.0
27
+ limit_test_batches: 1.0
28
+ num_sanity_val_steps: 2
29
+ log_grad_norm: true
30
+ log_param_norm: true
31
+ log_norm_every_n_steps: 50
32
+ data:
33
+ name: synth-u
34
+ data_folder: __DATA_ROOT__/datasets/synth-u
35
+ n_var: 1
36
+ seq_length: 128
37
+ normalize: false
38
+ train_split: 0.8
39
+ val_split: 0.1
40
+ model:
41
+ name: bridge
42
+ channels: 64
43
+ layers: 3
44
+ nheads: 8
45
+ dropout: 0.0
46
+ latent_dim: 32
47
+ num_latents: 16
48
+ latent_unit: 1
49
+ repre_emb_channels: 32
50
+ prototype_dim: 32
51
+ context_dim: 32
52
+ text_dim: 1024
53
+ fusion_type: gated_add
54
+ cond_drop_prob: 0.5
55
+ use_cfg: true
56
+ model_channels: 64
57
+ num_res_blocks: 2
58
+ attention_resolutions:
59
+ - 1
60
+ - 2
61
+ - 4
62
+ channel_mult:
63
+ - 1
64
+ - 2
65
+ - 4
66
+ - 4
67
+ num_heads: 8
68
+ num_head_channels: -1
69
+ use_scale_shift_norm: true
70
+ resblock_updown: true
71
+ use_spatial_transformer: true
72
+ transformer_depth: 1
73
+ diffusion_steps: 50
74
+ noise_schedule: quad
75
+ beta_start: 0.0001
76
+ beta_end: 0.5
77
+ bridge_type: vp
78
+ condition:
79
+ text:
80
+ enabled: true
81
+ input_dim: 1024
82
+ embedding_key: cap_emb
83
+ dropout: 0.0
84
+ cfg_scale: 1.0
85
+ attribute:
86
+ enabled: false
87
+ continuous_dim: 0
88
+ discrete_configs: []
89
+ output_dim: 128
90
+ dropout: 0.0
91
+ label:
92
+ enabled: false
93
+ num_classes: 10
94
+ output_dim: 64
95
+ fusion: concat
96
+ condition_dropout: 0.0
97
+ eval:
98
+ n_samples: 10
99
+ metrics:
100
+ - dtw
101
+ - wape
102
+ - ed
103
+ - crps
104
+ - acd
105
+ - sd
106
+ - kd
107
+ - mdd
108
+ - fid
109
+ - sfd
110
+ - jftsd
111
+ - prdc_f1
112
+ - joint_prdc_f1
113
+ - cttp
114
+ - disc_auc
115
+ batch_size: 128
116
+ save_samples: true
117
+ sampler: ddim
118
+ display_interval: 10
119
+ clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-u/model_configs.yaml
120
+ clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-u/clip_model_best.pth
121
+ cache_folder: __CACHE_ROOT__/bridge/synth-u/seed0
122
+ use_longalign: false
123
+ reference_split: train
124
+ disc_auc:
125
+ k_folds: 5
126
+ epochs: 5
127
+ batch_size: 128
128
+ lr: 0.001
129
+ weight_decay: 0.0001
130
+ hidden_channels: 128
131
+ num_layers: 3
132
+ dropout: 0.1
133
+ seed: 42
134
+ num_workers: 0
135
+ prdc:
136
+ k: 5
137
+ max_samples: 5000
138
+ seed: 0
139
+ metric: euclidean
140
+ backend: auto
141
+ knn_chunk_size: 4096
142
+ joint_enable: true
143
+ joint_weights:
144
+ ts: 1.0
145
+ text: 1.0
146
+ joint_normalize: standard
147
+ segment_classifier:
148
+ enable: false
149
+ segment_len: 128
150
+ n_segments: 3
151
+ seed: 42
152
+ acd_max_lag: 50
153
+ mdd_bins: 32
154
+ use_cache: false
155
+ cache_file: predictions_cache.pkl
156
+ viz:
157
+ enable: true
158
+ k_cases: 20
159
+ max_vars: 8
160
+ ncols: 5
161
+ seed: 42
162
+ dpi: 200
163
+ alpha: 0.25
164
+ figsize_per_subplot:
165
+ - 3.0
166
+ - 2.0
167
+ output_type: pdf
168
+ output_dir: ''
169
+ created_at: '2025-12-28T18:38:34.831322'
experiments/bridge/synth-u/seed0/results/expected_seed_metrics.json ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dtw": 6.354155136104681,
3
+ "wape": 88.65556005794502,
4
+ "ed": 9.25147834777832,
5
+ "crps": 0.5361806164317251,
6
+ "acd": 0.02120952121913433,
7
+ "sd": 0.17698199261192563,
8
+ "kd": 0.5880551470131783,
9
+ "mdd": 0.02806243896484374,
10
+ "fid": 47.42080353150922,
11
+ "sfd": 9.808309640175452,
12
+ "jftsd": 52.52638332382759,
13
+ "prdc_f1.f1": 0.14994884425918906,
14
+ "prdc_f1.precision": 0.5735,
15
+ "prdc_f1.recall": 0.08625,
16
+ "joint_prdc_f1.f1": 0.610718726016884,
17
+ "joint_prdc_f1.precision": 0.8145,
18
+ "joint_prdc_f1.recall": 0.4885,
19
+ "cttp": 23.847719635009767,
20
+ "disc_auc": 0.9995162499999999
21
+ }
experiments/bridge/synth-u/seed0/summary.json ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ {
2
+ "status": "completed",
3
+ "finished_at": "2025-12-28T19:47:20.050140",
4
+ "best_checkpoint": "checkpoints/finetune/best.ckpt"
5
+ }
experiments/bridge/synth-u/seed1/config.template.yaml ADDED
@@ -0,0 +1,169 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: synth-u_bridge
2
+ description: ''
3
+ seed: 1
4
+ device: cuda:0
5
+ train:
6
+ stages:
7
+ - name: finetune
8
+ epochs: 700
9
+ lr: 0.001
10
+ use_condition: true
11
+ freeze_modules: []
12
+ early_stopping_patience: 50
13
+ epochs: 700
14
+ batch_size: 256
15
+ lr: 0.001
16
+ weight_decay: 0.0001
17
+ scheduler: cosine
18
+ scheduler_params: {}
19
+ early_stopping_patience: 50
20
+ gradient_clip_val: 1.0
21
+ accumulate_grad_batches: 1
22
+ val_check_interval: 1.0
23
+ num_workers: 4
24
+ pin_memory: true
25
+ limit_train_batches: 1.0
26
+ limit_val_batches: 1.0
27
+ limit_test_batches: 1.0
28
+ num_sanity_val_steps: 2
29
+ log_grad_norm: true
30
+ log_param_norm: true
31
+ log_norm_every_n_steps: 50
32
+ data:
33
+ name: synth-u
34
+ data_folder: __DATA_ROOT__/datasets/synth-u
35
+ n_var: 1
36
+ seq_length: 128
37
+ normalize: false
38
+ train_split: 0.8
39
+ val_split: 0.1
40
+ model:
41
+ name: bridge
42
+ channels: 64
43
+ layers: 3
44
+ nheads: 8
45
+ dropout: 0.0
46
+ latent_dim: 32
47
+ num_latents: 16
48
+ latent_unit: 1
49
+ repre_emb_channels: 32
50
+ prototype_dim: 32
51
+ context_dim: 32
52
+ text_dim: 1024
53
+ fusion_type: gated_add
54
+ cond_drop_prob: 0.5
55
+ use_cfg: true
56
+ model_channels: 64
57
+ num_res_blocks: 2
58
+ attention_resolutions:
59
+ - 1
60
+ - 2
61
+ - 4
62
+ channel_mult:
63
+ - 1
64
+ - 2
65
+ - 4
66
+ - 4
67
+ num_heads: 8
68
+ num_head_channels: -1
69
+ use_scale_shift_norm: true
70
+ resblock_updown: true
71
+ use_spatial_transformer: true
72
+ transformer_depth: 1
73
+ diffusion_steps: 50
74
+ noise_schedule: quad
75
+ beta_start: 0.0001
76
+ beta_end: 0.5
77
+ bridge_type: vp
78
+ condition:
79
+ text:
80
+ enabled: true
81
+ input_dim: 1024
82
+ embedding_key: cap_emb
83
+ dropout: 0.0
84
+ cfg_scale: 1.0
85
+ attribute:
86
+ enabled: false
87
+ continuous_dim: 0
88
+ discrete_configs: []
89
+ output_dim: 128
90
+ dropout: 0.0
91
+ label:
92
+ enabled: false
93
+ num_classes: 10
94
+ output_dim: 64
95
+ fusion: concat
96
+ condition_dropout: 0.0
97
+ eval:
98
+ n_samples: 10
99
+ metrics:
100
+ - dtw
101
+ - wape
102
+ - ed
103
+ - crps
104
+ - acd
105
+ - sd
106
+ - kd
107
+ - mdd
108
+ - fid
109
+ - sfd
110
+ - jftsd
111
+ - prdc_f1
112
+ - joint_prdc_f1
113
+ - cttp
114
+ - disc_auc
115
+ batch_size: 128
116
+ save_samples: true
117
+ sampler: ddim
118
+ display_interval: 10
119
+ clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-u/model_configs.yaml
120
+ clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-u/clip_model_best.pth
121
+ cache_folder: __CACHE_ROOT__/bridge/synth-u/seed1
122
+ use_longalign: false
123
+ reference_split: train
124
+ disc_auc:
125
+ k_folds: 5
126
+ epochs: 5
127
+ batch_size: 128
128
+ lr: 0.001
129
+ weight_decay: 0.0001
130
+ hidden_channels: 128
131
+ num_layers: 3
132
+ dropout: 0.1
133
+ seed: 42
134
+ num_workers: 0
135
+ prdc:
136
+ k: 5
137
+ max_samples: 5000
138
+ seed: 0
139
+ metric: euclidean
140
+ backend: auto
141
+ knn_chunk_size: 4096
142
+ joint_enable: true
143
+ joint_weights:
144
+ ts: 1.0
145
+ text: 1.0
146
+ joint_normalize: standard
147
+ segment_classifier:
148
+ enable: false
149
+ segment_len: 128
150
+ n_segments: 3
151
+ seed: 42
152
+ acd_max_lag: 50
153
+ mdd_bins: 32
154
+ use_cache: false
155
+ cache_file: predictions_cache.pkl
156
+ viz:
157
+ enable: true
158
+ k_cases: 20
159
+ max_vars: 8
160
+ ncols: 5
161
+ seed: 42
162
+ dpi: 200
163
+ alpha: 0.25
164
+ figsize_per_subplot:
165
+ - 3.0
166
+ - 2.0
167
+ output_type: pdf
168
+ output_dir: ''
169
+ created_at: '2025-12-28T18:38:34.803045'
experiments/bridge/synth-u/seed1/results/expected_seed_metrics.json ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dtw": 5.753590787621783,
3
+ "wape": 85.46210875970107,
4
+ "ed": 8.889996612548828,
5
+ "crps": 0.5136260835656676,
6
+ "acd": 0.028903482481837273,
7
+ "sd": 0.022280193616036857,
8
+ "kd": 0.6368266785364254,
9
+ "mdd": 0.028317220052083324,
10
+ "fid": 48.50368331516401,
11
+ "sfd": 9.037110945980103,
12
+ "jftsd": 54.73198046870306,
13
+ "prdc_f1.f1": 0.1260411555203799,
14
+ "prdc_f1.precision": 0.56075,
15
+ "prdc_f1.recall": 0.071,
16
+ "joint_prdc_f1.f1": 0.5706327191739218,
17
+ "joint_prdc_f1.precision": 0.78725,
18
+ "joint_prdc_f1.recall": 0.4475,
19
+ "cttp": 22.753153106689453,
20
+ "disc_auc": 0.9995903125
21
+ }
experiments/bridge/synth-u/seed1/summary.json ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ {
2
+ "status": "completed",
3
+ "finished_at": "2025-12-28T19:17:30.004892",
4
+ "best_checkpoint": "checkpoints/finetune/best.ckpt"
5
+ }
experiments/bridge/synth-u/seed2/config.template.yaml ADDED
@@ -0,0 +1,169 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: synth-u_bridge
2
+ description: ''
3
+ seed: 2
4
+ device: cuda:0
5
+ train:
6
+ stages:
7
+ - name: finetune
8
+ epochs: 700
9
+ lr: 0.001
10
+ use_condition: true
11
+ freeze_modules: []
12
+ early_stopping_patience: 50
13
+ epochs: 700
14
+ batch_size: 256
15
+ lr: 0.001
16
+ weight_decay: 0.0001
17
+ scheduler: cosine
18
+ scheduler_params: {}
19
+ early_stopping_patience: 50
20
+ gradient_clip_val: 1.0
21
+ accumulate_grad_batches: 1
22
+ val_check_interval: 1.0
23
+ num_workers: 4
24
+ pin_memory: true
25
+ limit_train_batches: 1.0
26
+ limit_val_batches: 1.0
27
+ limit_test_batches: 1.0
28
+ num_sanity_val_steps: 2
29
+ log_grad_norm: true
30
+ log_param_norm: true
31
+ log_norm_every_n_steps: 50
32
+ data:
33
+ name: synth-u
34
+ data_folder: __DATA_ROOT__/datasets/synth-u
35
+ n_var: 1
36
+ seq_length: 128
37
+ normalize: false
38
+ train_split: 0.8
39
+ val_split: 0.1
40
+ model:
41
+ name: bridge
42
+ channels: 64
43
+ layers: 3
44
+ nheads: 8
45
+ dropout: 0.0
46
+ latent_dim: 32
47
+ num_latents: 16
48
+ latent_unit: 1
49
+ repre_emb_channels: 32
50
+ prototype_dim: 32
51
+ context_dim: 32
52
+ text_dim: 1024
53
+ fusion_type: gated_add
54
+ cond_drop_prob: 0.5
55
+ use_cfg: true
56
+ model_channels: 64
57
+ num_res_blocks: 2
58
+ attention_resolutions:
59
+ - 1
60
+ - 2
61
+ - 4
62
+ channel_mult:
63
+ - 1
64
+ - 2
65
+ - 4
66
+ - 4
67
+ num_heads: 8
68
+ num_head_channels: -1
69
+ use_scale_shift_norm: true
70
+ resblock_updown: true
71
+ use_spatial_transformer: true
72
+ transformer_depth: 1
73
+ diffusion_steps: 50
74
+ noise_schedule: quad
75
+ beta_start: 0.0001
76
+ beta_end: 0.5
77
+ bridge_type: vp
78
+ condition:
79
+ text:
80
+ enabled: true
81
+ input_dim: 1024
82
+ embedding_key: cap_emb
83
+ dropout: 0.0
84
+ cfg_scale: 1.0
85
+ attribute:
86
+ enabled: false
87
+ continuous_dim: 0
88
+ discrete_configs: []
89
+ output_dim: 128
90
+ dropout: 0.0
91
+ label:
92
+ enabled: false
93
+ num_classes: 10
94
+ output_dim: 64
95
+ fusion: concat
96
+ condition_dropout: 0.0
97
+ eval:
98
+ n_samples: 10
99
+ metrics:
100
+ - dtw
101
+ - wape
102
+ - ed
103
+ - crps
104
+ - acd
105
+ - sd
106
+ - kd
107
+ - mdd
108
+ - fid
109
+ - sfd
110
+ - jftsd
111
+ - prdc_f1
112
+ - joint_prdc_f1
113
+ - cttp
114
+ - disc_auc
115
+ batch_size: 128
116
+ save_samples: true
117
+ sampler: ddim
118
+ display_interval: 10
119
+ clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-u/model_configs.yaml
120
+ clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-u/clip_model_best.pth
121
+ cache_folder: __CACHE_ROOT__/bridge/synth-u/seed2
122
+ use_longalign: false
123
+ reference_split: train
124
+ disc_auc:
125
+ k_folds: 5
126
+ epochs: 5
127
+ batch_size: 128
128
+ lr: 0.001
129
+ weight_decay: 0.0001
130
+ hidden_channels: 128
131
+ num_layers: 3
132
+ dropout: 0.1
133
+ seed: 42
134
+ num_workers: 0
135
+ prdc:
136
+ k: 5
137
+ max_samples: 5000
138
+ seed: 0
139
+ metric: euclidean
140
+ backend: auto
141
+ knn_chunk_size: 4096
142
+ joint_enable: true
143
+ joint_weights:
144
+ ts: 1.0
145
+ text: 1.0
146
+ joint_normalize: standard
147
+ segment_classifier:
148
+ enable: false
149
+ segment_len: 128
150
+ n_segments: 3
151
+ seed: 42
152
+ acd_max_lag: 50
153
+ mdd_bins: 32
154
+ use_cache: false
155
+ cache_file: predictions_cache.pkl
156
+ viz:
157
+ enable: true
158
+ k_cases: 20
159
+ max_vars: 8
160
+ ncols: 5
161
+ seed: 42
162
+ dpi: 200
163
+ alpha: 0.25
164
+ figsize_per_subplot:
165
+ - 3.0
166
+ - 2.0
167
+ output_type: pdf
168
+ output_dir: ''
169
+ created_at: '2025-12-28T18:40:44.845855'
experiments/bridge/synth-u/seed2/results/expected_seed_metrics.json ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dtw": 5.530313082307284,
3
+ "wape": 82.54036397922455,
4
+ "ed": 8.522937805175781,
5
+ "crps": 0.5017315307045973,
6
+ "acd": 0.04407091811299324,
7
+ "sd": 0.0928177492640565,
8
+ "kd": 0.7001046294738782,
9
+ "mdd": 0.028313680013020828,
10
+ "fid": 54.61961133293579,
11
+ "sfd": 9.44118332955128,
12
+ "jftsd": 58.16182762137342,
13
+ "prdc_f1.f1": 0.1200266429840142,
14
+ "prdc_f1.precision": 0.6375,
15
+ "prdc_f1.recall": 0.06625,
16
+ "joint_prdc_f1.f1": 0.6256576958525346,
17
+ "joint_prdc_f1.precision": 0.86675,
18
+ "joint_prdc_f1.recall": 0.4895,
19
+ "cttp": 22.37494738769531,
20
+ "disc_auc": 0.99950046875
21
+ }
experiments/bridge/synth-u/seed2/summary.json ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ {
2
+ "status": "completed",
3
+ "finished_at": "2025-12-28T20:17:12.870391",
4
+ "best_checkpoint": "checkpoints/finetune/best.ckpt"
5
+ }
experiments/diffusets/synth-m/seed0/config.template.yaml ADDED
@@ -0,0 +1,156 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: synth-m_diffusets
2
+ description: ''
3
+ seed: 0
4
+ device: cuda:0
5
+ train:
6
+ stages:
7
+ - name: vae_pretrain
8
+ epochs: 200
9
+ lr: 0.001
10
+ use_condition: false
11
+ freeze_modules: []
12
+ early_stopping_patience: 50
13
+ - name: finetune
14
+ epochs: 500
15
+ lr: 0.001
16
+ use_condition: true
17
+ freeze_modules: []
18
+ load_from_stage: vae_pretrain
19
+ early_stopping_patience: 50
20
+ epochs: 700
21
+ batch_size: 256
22
+ lr: 0.001
23
+ weight_decay: 0.0001
24
+ scheduler: cosine
25
+ scheduler_params: {}
26
+ early_stopping_patience: 50
27
+ gradient_clip_val: 1.0
28
+ accumulate_grad_batches: 1
29
+ val_check_interval: 1.0
30
+ num_workers: 4
31
+ pin_memory: true
32
+ limit_train_batches: 1.0
33
+ limit_val_batches: 1.0
34
+ limit_test_batches: 1.0
35
+ num_sanity_val_steps: 2
36
+ log_grad_norm: true
37
+ log_param_norm: true
38
+ log_norm_every_n_steps: 50
39
+ data:
40
+ name: synth-m
41
+ data_folder: __DATA_ROOT__/datasets/synth-m
42
+ n_var: 2
43
+ seq_length: 128
44
+ normalize: false
45
+ train_split: 0.8
46
+ val_split: 0.1
47
+ model:
48
+ name: diffusets
49
+ channels: 64
50
+ layers: 3
51
+ nheads: 8
52
+ dropout: 0.1
53
+ latent_channels: 4
54
+ diffusion_steps: 50
55
+ noise_schedule: quad
56
+ beta_start: 0.0001
57
+ beta_end: 0.5
58
+ kernel_size: 3
59
+ num_levels: 5
60
+ kld_weight: 1.0
61
+ kld_annealing: true
62
+ kld_annealing_type: linear
63
+ kld_warmup_epochs: 300
64
+ kld_start_weight: 0.01
65
+ condition:
66
+ text:
67
+ enabled: true
68
+ input_dim: 1024
69
+ embedding_key: cap_emb
70
+ dropout: 0.0
71
+ cfg_scale: 1.0
72
+ attribute:
73
+ enabled: false
74
+ continuous_dim: 0
75
+ discrete_configs: []
76
+ output_dim: 128
77
+ dropout: 0.0
78
+ label:
79
+ enabled: false
80
+ num_classes: 10
81
+ output_dim: 64
82
+ fusion: concat
83
+ condition_dropout: 0.0
84
+ eval:
85
+ n_samples: 10
86
+ metrics:
87
+ - dtw
88
+ - wape
89
+ - ed
90
+ - crps
91
+ - acd
92
+ - sd
93
+ - kd
94
+ - mdd
95
+ - fid
96
+ - sfd
97
+ - jftsd
98
+ - prdc_f1
99
+ - joint_prdc_f1
100
+ - cttp
101
+ - disc_auc
102
+ batch_size: 128
103
+ save_samples: true
104
+ sampler: ddim
105
+ display_interval: 10
106
+ clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-m/model_configs.yaml
107
+ clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-m/clip_model_best.pth
108
+ cache_folder: __CACHE_ROOT__/diffusets/synth-m/seed0
109
+ use_longalign: false
110
+ reference_split: train
111
+ disc_auc:
112
+ k_folds: 5
113
+ epochs: 5
114
+ batch_size: 128
115
+ lr: 0.001
116
+ weight_decay: 0.0001
117
+ hidden_channels: 128
118
+ num_layers: 3
119
+ dropout: 0.1
120
+ seed: 42
121
+ num_workers: 0
122
+ prdc:
123
+ k: 5
124
+ max_samples: 5000
125
+ seed: 0
126
+ metric: euclidean
127
+ backend: auto
128
+ knn_chunk_size: 4096
129
+ joint_enable: true
130
+ joint_weights:
131
+ ts: 1.0
132
+ text: 1.0
133
+ joint_normalize: standard
134
+ segment_classifier:
135
+ enable: false
136
+ segment_len: 128
137
+ n_segments: 3
138
+ seed: 42
139
+ acd_max_lag: 50
140
+ mdd_bins: 32
141
+ use_cache: false
142
+ cache_file: predictions_cache.pkl
143
+ viz:
144
+ enable: true
145
+ k_cases: 20
146
+ max_vars: 8
147
+ ncols: 5
148
+ seed: 42
149
+ dpi: 200
150
+ alpha: 0.25
151
+ figsize_per_subplot:
152
+ - 3.0
153
+ - 2.0
154
+ output_type: pdf
155
+ output_dir: ''
156
+ created_at: '2025-12-28T15:52:00.016312'
experiments/diffusets/synth-m/seed0/results/expected_seed_metrics.json ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dtw": 11.140782107069837,
3
+ "wape": 89.60107209439903,
4
+ "ed": 13.401970443725586,
5
+ "crps": 0.48672253132439686,
6
+ "acd": 0.03553586080670357,
7
+ "sd": 0.16719546573763044,
8
+ "kd": 0.07568088321032929,
9
+ "mdd": 0.023467702229817703,
10
+ "fid": 42.76190316351483,
11
+ "sfd": 85.36533673707893,
12
+ "jftsd": 52.20228797971308,
13
+ "prdc_f1.f1": 0.2579129526462396,
14
+ "prdc_f1.precision": 0.5495,
15
+ "prdc_f1.recall": 0.1685,
16
+ "joint_prdc_f1.f1": 0.305841307814992,
17
+ "joint_prdc_f1.precision": 0.3625,
18
+ "joint_prdc_f1.recall": 0.2645,
19
+ "cttp": 16.67788977050781,
20
+ "disc_auc": 0.9999359375000001
21
+ }
experiments/diffusets/synth-m/seed0/summary.json ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ {
2
+ "status": "completed",
3
+ "finished_at": "2025-12-28T16:24:00.801438",
4
+ "best_checkpoint": "checkpoints/finetune/best.ckpt"
5
+ }
experiments/diffusets/synth-m/seed1/config.template.yaml ADDED
@@ -0,0 +1,156 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: synth-m_diffusets
2
+ description: ''
3
+ seed: 1
4
+ device: cuda:0
5
+ train:
6
+ stages:
7
+ - name: vae_pretrain
8
+ epochs: 200
9
+ lr: 0.001
10
+ use_condition: false
11
+ freeze_modules: []
12
+ early_stopping_patience: 50
13
+ - name: finetune
14
+ epochs: 500
15
+ lr: 0.001
16
+ use_condition: true
17
+ freeze_modules: []
18
+ load_from_stage: vae_pretrain
19
+ early_stopping_patience: 50
20
+ epochs: 700
21
+ batch_size: 256
22
+ lr: 0.001
23
+ weight_decay: 0.0001
24
+ scheduler: cosine
25
+ scheduler_params: {}
26
+ early_stopping_patience: 50
27
+ gradient_clip_val: 1.0
28
+ accumulate_grad_batches: 1
29
+ val_check_interval: 1.0
30
+ num_workers: 4
31
+ pin_memory: true
32
+ limit_train_batches: 1.0
33
+ limit_val_batches: 1.0
34
+ limit_test_batches: 1.0
35
+ num_sanity_val_steps: 2
36
+ log_grad_norm: true
37
+ log_param_norm: true
38
+ log_norm_every_n_steps: 50
39
+ data:
40
+ name: synth-m
41
+ data_folder: __DATA_ROOT__/datasets/synth-m
42
+ n_var: 2
43
+ seq_length: 128
44
+ normalize: false
45
+ train_split: 0.8
46
+ val_split: 0.1
47
+ model:
48
+ name: diffusets
49
+ channels: 64
50
+ layers: 3
51
+ nheads: 8
52
+ dropout: 0.1
53
+ latent_channels: 4
54
+ diffusion_steps: 50
55
+ noise_schedule: quad
56
+ beta_start: 0.0001
57
+ beta_end: 0.5
58
+ kernel_size: 3
59
+ num_levels: 5
60
+ kld_weight: 1.0
61
+ kld_annealing: true
62
+ kld_annealing_type: linear
63
+ kld_warmup_epochs: 300
64
+ kld_start_weight: 0.01
65
+ condition:
66
+ text:
67
+ enabled: true
68
+ input_dim: 1024
69
+ embedding_key: cap_emb
70
+ dropout: 0.0
71
+ cfg_scale: 1.0
72
+ attribute:
73
+ enabled: false
74
+ continuous_dim: 0
75
+ discrete_configs: []
76
+ output_dim: 128
77
+ dropout: 0.0
78
+ label:
79
+ enabled: false
80
+ num_classes: 10
81
+ output_dim: 64
82
+ fusion: concat
83
+ condition_dropout: 0.0
84
+ eval:
85
+ n_samples: 10
86
+ metrics:
87
+ - dtw
88
+ - wape
89
+ - ed
90
+ - crps
91
+ - acd
92
+ - sd
93
+ - kd
94
+ - mdd
95
+ - fid
96
+ - sfd
97
+ - jftsd
98
+ - prdc_f1
99
+ - joint_prdc_f1
100
+ - cttp
101
+ - disc_auc
102
+ batch_size: 128
103
+ save_samples: true
104
+ sampler: ddim
105
+ display_interval: 10
106
+ clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-m/model_configs.yaml
107
+ clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-m/clip_model_best.pth
108
+ cache_folder: __CACHE_ROOT__/diffusets/synth-m/seed1
109
+ use_longalign: false
110
+ reference_split: train
111
+ disc_auc:
112
+ k_folds: 5
113
+ epochs: 5
114
+ batch_size: 128
115
+ lr: 0.001
116
+ weight_decay: 0.0001
117
+ hidden_channels: 128
118
+ num_layers: 3
119
+ dropout: 0.1
120
+ seed: 42
121
+ num_workers: 0
122
+ prdc:
123
+ k: 5
124
+ max_samples: 5000
125
+ seed: 0
126
+ metric: euclidean
127
+ backend: auto
128
+ knn_chunk_size: 4096
129
+ joint_enable: true
130
+ joint_weights:
131
+ ts: 1.0
132
+ text: 1.0
133
+ joint_normalize: standard
134
+ segment_classifier:
135
+ enable: false
136
+ segment_len: 128
137
+ n_segments: 3
138
+ seed: 42
139
+ acd_max_lag: 50
140
+ mdd_bins: 32
141
+ use_cache: false
142
+ cache_file: predictions_cache.pkl
143
+ viz:
144
+ enable: true
145
+ k_cases: 20
146
+ max_vars: 8
147
+ ncols: 5
148
+ seed: 42
149
+ dpi: 200
150
+ alpha: 0.25
151
+ figsize_per_subplot:
152
+ - 3.0
153
+ - 2.0
154
+ output_type: pdf
155
+ output_dir: ''
156
+ created_at: '2025-12-28T15:55:48.457922'
experiments/diffusets/synth-m/seed1/results/expected_seed_metrics.json ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dtw": 10.334665453160229,
3
+ "wape": 85.56766340774767,
4
+ "ed": 12.78905828857422,
5
+ "crps": 0.4503502460532916,
6
+ "acd": 0.025928545743227005,
7
+ "sd": 0.0435979218540752,
8
+ "kd": 0.08138287083778217,
9
+ "mdd": 0.021443400065104163,
10
+ "fid": 33.25484026734239,
11
+ "sfd": 79.12128260221301,
12
+ "jftsd": 39.5227926010414,
13
+ "prdc_f1.f1": 0.4241700302724521,
14
+ "prdc_f1.precision": 0.6835,
15
+ "prdc_f1.recall": 0.3075,
16
+ "joint_prdc_f1.f1": 0.5467748411978223,
17
+ "joint_prdc_f1.precision": 0.59925,
18
+ "joint_prdc_f1.recall": 0.50275,
19
+ "cttp": 21.062711242675782,
20
+ "disc_auc": 0.9997506249999999
21
+ }
experiments/diffusets/synth-m/seed1/summary.json ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ {
2
+ "status": "completed",
3
+ "finished_at": "2025-12-28T16:57:34.102696",
4
+ "best_checkpoint": "checkpoints/finetune/best.ckpt"
5
+ }
experiments/diffusets/synth-m/seed2/config.template.yaml ADDED
@@ -0,0 +1,156 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: synth-m_diffusets
2
+ description: ''
3
+ seed: 2
4
+ device: cuda:0
5
+ train:
6
+ stages:
7
+ - name: vae_pretrain
8
+ epochs: 200
9
+ lr: 0.001
10
+ use_condition: false
11
+ freeze_modules: []
12
+ early_stopping_patience: 50
13
+ - name: finetune
14
+ epochs: 500
15
+ lr: 0.001
16
+ use_condition: true
17
+ freeze_modules: []
18
+ load_from_stage: vae_pretrain
19
+ early_stopping_patience: 50
20
+ epochs: 700
21
+ batch_size: 256
22
+ lr: 0.001
23
+ weight_decay: 0.0001
24
+ scheduler: cosine
25
+ scheduler_params: {}
26
+ early_stopping_patience: 50
27
+ gradient_clip_val: 1.0
28
+ accumulate_grad_batches: 1
29
+ val_check_interval: 1.0
30
+ num_workers: 4
31
+ pin_memory: true
32
+ limit_train_batches: 1.0
33
+ limit_val_batches: 1.0
34
+ limit_test_batches: 1.0
35
+ num_sanity_val_steps: 2
36
+ log_grad_norm: true
37
+ log_param_norm: true
38
+ log_norm_every_n_steps: 50
39
+ data:
40
+ name: synth-m
41
+ data_folder: __DATA_ROOT__/datasets/synth-m
42
+ n_var: 2
43
+ seq_length: 128
44
+ normalize: false
45
+ train_split: 0.8
46
+ val_split: 0.1
47
+ model:
48
+ name: diffusets
49
+ channels: 64
50
+ layers: 3
51
+ nheads: 8
52
+ dropout: 0.1
53
+ latent_channels: 4
54
+ diffusion_steps: 50
55
+ noise_schedule: quad
56
+ beta_start: 0.0001
57
+ beta_end: 0.5
58
+ kernel_size: 3
59
+ num_levels: 5
60
+ kld_weight: 1.0
61
+ kld_annealing: true
62
+ kld_annealing_type: linear
63
+ kld_warmup_epochs: 300
64
+ kld_start_weight: 0.01
65
+ condition:
66
+ text:
67
+ enabled: true
68
+ input_dim: 1024
69
+ embedding_key: cap_emb
70
+ dropout: 0.0
71
+ cfg_scale: 1.0
72
+ attribute:
73
+ enabled: false
74
+ continuous_dim: 0
75
+ discrete_configs: []
76
+ output_dim: 128
77
+ dropout: 0.0
78
+ label:
79
+ enabled: false
80
+ num_classes: 10
81
+ output_dim: 64
82
+ fusion: concat
83
+ condition_dropout: 0.0
84
+ eval:
85
+ n_samples: 10
86
+ metrics:
87
+ - dtw
88
+ - wape
89
+ - ed
90
+ - crps
91
+ - acd
92
+ - sd
93
+ - kd
94
+ - mdd
95
+ - fid
96
+ - sfd
97
+ - jftsd
98
+ - prdc_f1
99
+ - joint_prdc_f1
100
+ - cttp
101
+ - disc_auc
102
+ batch_size: 128
103
+ save_samples: true
104
+ sampler: ddim
105
+ display_interval: 10
106
+ clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-m/model_configs.yaml
107
+ clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-m/clip_model_best.pth
108
+ cache_folder: __CACHE_ROOT__/diffusets/synth-m/seed2
109
+ use_longalign: false
110
+ reference_split: train
111
+ disc_auc:
112
+ k_folds: 5
113
+ epochs: 5
114
+ batch_size: 128
115
+ lr: 0.001
116
+ weight_decay: 0.0001
117
+ hidden_channels: 128
118
+ num_layers: 3
119
+ dropout: 0.1
120
+ seed: 42
121
+ num_workers: 0
122
+ prdc:
123
+ k: 5
124
+ max_samples: 5000
125
+ seed: 0
126
+ metric: euclidean
127
+ backend: auto
128
+ knn_chunk_size: 4096
129
+ joint_enable: true
130
+ joint_weights:
131
+ ts: 1.0
132
+ text: 1.0
133
+ joint_normalize: standard
134
+ segment_classifier:
135
+ enable: false
136
+ segment_len: 128
137
+ n_segments: 3
138
+ seed: 42
139
+ acd_max_lag: 50
140
+ mdd_bins: 32
141
+ use_cache: false
142
+ cache_file: predictions_cache.pkl
143
+ viz:
144
+ enable: true
145
+ k_cases: 20
146
+ max_vars: 8
147
+ ncols: 5
148
+ seed: 42
149
+ dpi: 200
150
+ alpha: 0.25
151
+ figsize_per_subplot:
152
+ - 3.0
153
+ - 2.0
154
+ output_type: pdf
155
+ output_dir: ''
156
+ created_at: '2025-12-28T16:18:09.179981'
experiments/diffusets/synth-m/seed2/results/expected_seed_metrics.json ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dtw": 9.317690117480163,
3
+ "wape": 81.66055296593296,
4
+ "ed": 12.043457717895508,
5
+ "crps": 0.44030666400529395,
6
+ "acd": 0.06254220008850098,
7
+ "sd": 0.04333788460742726,
8
+ "kd": 0.3218432933887936,
9
+ "mdd": 0.01764912923177083,
10
+ "fid": 30.94725361796202,
11
+ "sfd": 44.83966561329548,
12
+ "jftsd": 34.4846521694688,
13
+ "prdc_f1.f1": 0.4201174869853289,
14
+ "prdc_f1.precision": 0.76725,
15
+ "prdc_f1.recall": 0.28925,
16
+ "joint_prdc_f1.f1": 0.7147140649149922,
17
+ "joint_prdc_f1.precision": 0.82575,
18
+ "joint_prdc_f1.recall": 0.63,
19
+ "cttp": 25.972833389282226,
20
+ "disc_auc": 0.9997812500000001
21
+ }
experiments/diffusets/synth-m/seed2/summary.json ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ {
2
+ "status": "completed",
3
+ "finished_at": "2025-12-28T17:31:42.961739",
4
+ "best_checkpoint": "checkpoints/finetune/best.ckpt"
5
+ }
experiments/diffusets/synth-u/seed0/config.template.yaml ADDED
@@ -0,0 +1,156 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: synth-u_diffusets
2
+ description: ''
3
+ seed: 0
4
+ device: cuda:0
5
+ train:
6
+ stages:
7
+ - name: vae_pretrain
8
+ epochs: 200
9
+ lr: 0.001
10
+ use_condition: false
11
+ freeze_modules: []
12
+ early_stopping_patience: 50
13
+ - name: finetune
14
+ epochs: 500
15
+ lr: 0.001
16
+ use_condition: true
17
+ freeze_modules: []
18
+ load_from_stage: vae_pretrain
19
+ early_stopping_patience: 50
20
+ epochs: 700
21
+ batch_size: 256
22
+ lr: 0.001
23
+ weight_decay: 0.0001
24
+ scheduler: cosine
25
+ scheduler_params: {}
26
+ early_stopping_patience: 50
27
+ gradient_clip_val: 1.0
28
+ accumulate_grad_batches: 1
29
+ val_check_interval: 1.0
30
+ num_workers: 4
31
+ pin_memory: true
32
+ limit_train_batches: 1.0
33
+ limit_val_batches: 1.0
34
+ limit_test_batches: 1.0
35
+ num_sanity_val_steps: 2
36
+ log_grad_norm: true
37
+ log_param_norm: true
38
+ log_norm_every_n_steps: 50
39
+ data:
40
+ name: synth-u
41
+ data_folder: __DATA_ROOT__/datasets/synth-u
42
+ n_var: 1
43
+ seq_length: 128
44
+ normalize: false
45
+ train_split: 0.8
46
+ val_split: 0.1
47
+ model:
48
+ name: diffusets
49
+ channels: 64
50
+ layers: 3
51
+ nheads: 8
52
+ dropout: 0.1
53
+ latent_channels: 4
54
+ diffusion_steps: 50
55
+ noise_schedule: quad
56
+ beta_start: 0.0001
57
+ beta_end: 0.5
58
+ kernel_size: 3
59
+ num_levels: 5
60
+ kld_weight: 1.0
61
+ kld_annealing: true
62
+ kld_annealing_type: linear
63
+ kld_warmup_epochs: 300
64
+ kld_start_weight: 0.01
65
+ condition:
66
+ text:
67
+ enabled: true
68
+ input_dim: 1024
69
+ embedding_key: cap_emb
70
+ dropout: 0.0
71
+ cfg_scale: 1.0
72
+ attribute:
73
+ enabled: false
74
+ continuous_dim: 0
75
+ discrete_configs: []
76
+ output_dim: 128
77
+ dropout: 0.0
78
+ label:
79
+ enabled: false
80
+ num_classes: 10
81
+ output_dim: 64
82
+ fusion: concat
83
+ condition_dropout: 0.0
84
+ eval:
85
+ n_samples: 10
86
+ metrics:
87
+ - dtw
88
+ - wape
89
+ - ed
90
+ - crps
91
+ - acd
92
+ - sd
93
+ - kd
94
+ - mdd
95
+ - fid
96
+ - sfd
97
+ - jftsd
98
+ - prdc_f1
99
+ - joint_prdc_f1
100
+ - cttp
101
+ - disc_auc
102
+ batch_size: 128
103
+ save_samples: true
104
+ sampler: ddim
105
+ display_interval: 10
106
+ clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-u/model_configs.yaml
107
+ clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-u/clip_model_best.pth
108
+ cache_folder: __CACHE_ROOT__/diffusets/synth-u/seed0
109
+ use_longalign: false
110
+ reference_split: train
111
+ disc_auc:
112
+ k_folds: 5
113
+ epochs: 5
114
+ batch_size: 128
115
+ lr: 0.001
116
+ weight_decay: 0.0001
117
+ hidden_channels: 128
118
+ num_layers: 3
119
+ dropout: 0.1
120
+ seed: 42
121
+ num_workers: 0
122
+ prdc:
123
+ k: 5
124
+ max_samples: 5000
125
+ seed: 0
126
+ metric: euclidean
127
+ backend: auto
128
+ knn_chunk_size: 4096
129
+ joint_enable: true
130
+ joint_weights:
131
+ ts: 1.0
132
+ text: 1.0
133
+ joint_normalize: standard
134
+ segment_classifier:
135
+ enable: false
136
+ segment_len: 128
137
+ n_segments: 3
138
+ seed: 42
139
+ acd_max_lag: 50
140
+ mdd_bins: 32
141
+ use_cache: false
142
+ cache_file: predictions_cache.pkl
143
+ viz:
144
+ enable: true
145
+ k_cases: 20
146
+ max_vars: 8
147
+ ncols: 5
148
+ seed: 42
149
+ dpi: 200
150
+ alpha: 0.25
151
+ figsize_per_subplot:
152
+ - 3.0
153
+ - 2.0
154
+ output_type: pdf
155
+ output_dir: ''
156
+ created_at: '2025-12-28T16:18:09.184954'
experiments/diffusets/synth-u/seed0/results/expected_seed_metrics.json ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dtw": 6.532428591720535,
3
+ "wape": 87.80967145497634,
4
+ "ed": 9.246404472351074,
5
+ "crps": 0.4526067496610641,
6
+ "acd": 0.0433688648045063,
7
+ "sd": 0.2959921614171813,
8
+ "kd": 1.8287045933291228,
9
+ "mdd": 0.0260916748046875,
10
+ "fid": 51.36839796221051,
11
+ "sfd": 54.6720668718775,
12
+ "jftsd": 58.86045523270212,
13
+ "prdc_f1.f1": 0.3003663853727145,
14
+ "prdc_f1.precision": 0.4955,
15
+ "prdc_f1.recall": 0.2155,
16
+ "joint_prdc_f1.f1": 0.5770832266325224,
17
+ "joint_prdc_f1.precision": 0.657,
18
+ "joint_prdc_f1.recall": 0.5145,
19
+ "cttp": 19.388934158325196,
20
+ "disc_auc": 0.999903125
21
+ }
experiments/diffusets/synth-u/seed0/summary.json ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ {
2
+ "status": "completed",
3
+ "finished_at": "2025-12-28T17:39:02.597689",
4
+ "best_checkpoint": "checkpoints/finetune/best.ckpt"
5
+ }
experiments/diffusets/synth-u/seed1/config.template.yaml ADDED
@@ -0,0 +1,156 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: synth-u_diffusets
2
+ description: ''
3
+ seed: 1
4
+ device: cuda:0
5
+ train:
6
+ stages:
7
+ - name: vae_pretrain
8
+ epochs: 200
9
+ lr: 0.001
10
+ use_condition: false
11
+ freeze_modules: []
12
+ early_stopping_patience: 50
13
+ - name: finetune
14
+ epochs: 500
15
+ lr: 0.001
16
+ use_condition: true
17
+ freeze_modules: []
18
+ load_from_stage: vae_pretrain
19
+ early_stopping_patience: 50
20
+ epochs: 700
21
+ batch_size: 256
22
+ lr: 0.001
23
+ weight_decay: 0.0001
24
+ scheduler: cosine
25
+ scheduler_params: {}
26
+ early_stopping_patience: 50
27
+ gradient_clip_val: 1.0
28
+ accumulate_grad_batches: 1
29
+ val_check_interval: 1.0
30
+ num_workers: 4
31
+ pin_memory: true
32
+ limit_train_batches: 1.0
33
+ limit_val_batches: 1.0
34
+ limit_test_batches: 1.0
35
+ num_sanity_val_steps: 2
36
+ log_grad_norm: true
37
+ log_param_norm: true
38
+ log_norm_every_n_steps: 50
39
+ data:
40
+ name: synth-u
41
+ data_folder: __DATA_ROOT__/datasets/synth-u
42
+ n_var: 1
43
+ seq_length: 128
44
+ normalize: false
45
+ train_split: 0.8
46
+ val_split: 0.1
47
+ model:
48
+ name: diffusets
49
+ channels: 64
50
+ layers: 3
51
+ nheads: 8
52
+ dropout: 0.1
53
+ latent_channels: 4
54
+ diffusion_steps: 50
55
+ noise_schedule: quad
56
+ beta_start: 0.0001
57
+ beta_end: 0.5
58
+ kernel_size: 3
59
+ num_levels: 5
60
+ kld_weight: 1.0
61
+ kld_annealing: true
62
+ kld_annealing_type: linear
63
+ kld_warmup_epochs: 300
64
+ kld_start_weight: 0.01
65
+ condition:
66
+ text:
67
+ enabled: true
68
+ input_dim: 1024
69
+ embedding_key: cap_emb
70
+ dropout: 0.0
71
+ cfg_scale: 1.0
72
+ attribute:
73
+ enabled: false
74
+ continuous_dim: 0
75
+ discrete_configs: []
76
+ output_dim: 128
77
+ dropout: 0.0
78
+ label:
79
+ enabled: false
80
+ num_classes: 10
81
+ output_dim: 64
82
+ fusion: concat
83
+ condition_dropout: 0.0
84
+ eval:
85
+ n_samples: 10
86
+ metrics:
87
+ - dtw
88
+ - wape
89
+ - ed
90
+ - crps
91
+ - acd
92
+ - sd
93
+ - kd
94
+ - mdd
95
+ - fid
96
+ - sfd
97
+ - jftsd
98
+ - prdc_f1
99
+ - joint_prdc_f1
100
+ - cttp
101
+ - disc_auc
102
+ batch_size: 128
103
+ save_samples: true
104
+ sampler: ddim
105
+ display_interval: 10
106
+ clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-u/model_configs.yaml
107
+ clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-u/clip_model_best.pth
108
+ cache_folder: __CACHE_ROOT__/diffusets/synth-u/seed1
109
+ use_longalign: false
110
+ reference_split: train
111
+ disc_auc:
112
+ k_folds: 5
113
+ epochs: 5
114
+ batch_size: 128
115
+ lr: 0.001
116
+ weight_decay: 0.0001
117
+ hidden_channels: 128
118
+ num_layers: 3
119
+ dropout: 0.1
120
+ seed: 42
121
+ num_workers: 0
122
+ prdc:
123
+ k: 5
124
+ max_samples: 5000
125
+ seed: 0
126
+ metric: euclidean
127
+ backend: auto
128
+ knn_chunk_size: 4096
129
+ joint_enable: true
130
+ joint_weights:
131
+ ts: 1.0
132
+ text: 1.0
133
+ joint_normalize: standard
134
+ segment_classifier:
135
+ enable: false
136
+ segment_len: 128
137
+ n_segments: 3
138
+ seed: 42
139
+ acd_max_lag: 50
140
+ mdd_bins: 32
141
+ use_cache: false
142
+ cache_file: predictions_cache.pkl
143
+ viz:
144
+ enable: true
145
+ k_cases: 20
146
+ max_vars: 8
147
+ ncols: 5
148
+ seed: 42
149
+ dpi: 200
150
+ alpha: 0.25
151
+ figsize_per_subplot:
152
+ - 3.0
153
+ - 2.0
154
+ output_type: pdf
155
+ output_dir: ''
156
+ created_at: '2025-12-28T16:18:09.185009'
experiments/diffusets/synth-u/seed1/results/expected_seed_metrics.json ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dtw": 4.954761916935516,
3
+ "wape": 81.29904120650428,
4
+ "ed": 8.348238151550293,
5
+ "crps": 0.4236568123287523,
6
+ "acd": 0.05647331476211548,
7
+ "sd": 0.031882725015297264,
8
+ "kd": 0.2726460478386823,
9
+ "mdd": 0.01779575602213541,
10
+ "fid": 33.97250146771219,
11
+ "sfd": 7.695950143207412,
12
+ "jftsd": 37.68384150873891,
13
+ "prdc_f1.f1": 0.33002626143405134,
14
+ "prdc_f1.precision": 0.62275,
15
+ "prdc_f1.recall": 0.2245,
16
+ "joint_prdc_f1.f1": 0.8377482661944814,
17
+ "joint_prdc_f1.precision": 0.93625,
18
+ "joint_prdc_f1.recall": 0.758,
19
+ "cttp": 27.93660757446289,
20
+ "disc_auc": 0.9999784375
21
+ }
experiments/diffusets/synth-u/seed1/summary.json ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ {
2
+ "status": "completed",
3
+ "finished_at": "2025-12-28T17:34:41.687120",
4
+ "best_checkpoint": "checkpoints/finetune/best.ckpt"
5
+ }
experiments/diffusets/synth-u/seed2/config.template.yaml ADDED
@@ -0,0 +1,156 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: synth-u_diffusets
2
+ description: ''
3
+ seed: 2
4
+ device: cuda:0
5
+ train:
6
+ stages:
7
+ - name: vae_pretrain
8
+ epochs: 200
9
+ lr: 0.001
10
+ use_condition: false
11
+ freeze_modules: []
12
+ early_stopping_patience: 50
13
+ - name: finetune
14
+ epochs: 500
15
+ lr: 0.001
16
+ use_condition: true
17
+ freeze_modules: []
18
+ load_from_stage: vae_pretrain
19
+ early_stopping_patience: 50
20
+ epochs: 700
21
+ batch_size: 256
22
+ lr: 0.001
23
+ weight_decay: 0.0001
24
+ scheduler: cosine
25
+ scheduler_params: {}
26
+ early_stopping_patience: 50
27
+ gradient_clip_val: 1.0
28
+ accumulate_grad_batches: 1
29
+ val_check_interval: 1.0
30
+ num_workers: 4
31
+ pin_memory: true
32
+ limit_train_batches: 1.0
33
+ limit_val_batches: 1.0
34
+ limit_test_batches: 1.0
35
+ num_sanity_val_steps: 2
36
+ log_grad_norm: true
37
+ log_param_norm: true
38
+ log_norm_every_n_steps: 50
39
+ data:
40
+ name: synth-u
41
+ data_folder: __DATA_ROOT__/datasets/synth-u
42
+ n_var: 1
43
+ seq_length: 128
44
+ normalize: false
45
+ train_split: 0.8
46
+ val_split: 0.1
47
+ model:
48
+ name: diffusets
49
+ channels: 64
50
+ layers: 3
51
+ nheads: 8
52
+ dropout: 0.1
53
+ latent_channels: 4
54
+ diffusion_steps: 50
55
+ noise_schedule: quad
56
+ beta_start: 0.0001
57
+ beta_end: 0.5
58
+ kernel_size: 3
59
+ num_levels: 5
60
+ kld_weight: 1.0
61
+ kld_annealing: true
62
+ kld_annealing_type: linear
63
+ kld_warmup_epochs: 300
64
+ kld_start_weight: 0.01
65
+ condition:
66
+ text:
67
+ enabled: true
68
+ input_dim: 1024
69
+ embedding_key: cap_emb
70
+ dropout: 0.0
71
+ cfg_scale: 1.0
72
+ attribute:
73
+ enabled: false
74
+ continuous_dim: 0
75
+ discrete_configs: []
76
+ output_dim: 128
77
+ dropout: 0.0
78
+ label:
79
+ enabled: false
80
+ num_classes: 10
81
+ output_dim: 64
82
+ fusion: concat
83
+ condition_dropout: 0.0
84
+ eval:
85
+ n_samples: 10
86
+ metrics:
87
+ - dtw
88
+ - wape
89
+ - ed
90
+ - crps
91
+ - acd
92
+ - sd
93
+ - kd
94
+ - mdd
95
+ - fid
96
+ - sfd
97
+ - jftsd
98
+ - prdc_f1
99
+ - joint_prdc_f1
100
+ - cttp
101
+ - disc_auc
102
+ batch_size: 128
103
+ save_samples: true
104
+ sampler: ddim
105
+ display_interval: 10
106
+ clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-u/model_configs.yaml
107
+ clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-u/clip_model_best.pth
108
+ cache_folder: __CACHE_ROOT__/diffusets/synth-u/seed2
109
+ use_longalign: false
110
+ reference_split: train
111
+ disc_auc:
112
+ k_folds: 5
113
+ epochs: 5
114
+ batch_size: 128
115
+ lr: 0.001
116
+ weight_decay: 0.0001
117
+ hidden_channels: 128
118
+ num_layers: 3
119
+ dropout: 0.1
120
+ seed: 42
121
+ num_workers: 0
122
+ prdc:
123
+ k: 5
124
+ max_samples: 5000
125
+ seed: 0
126
+ metric: euclidean
127
+ backend: auto
128
+ knn_chunk_size: 4096
129
+ joint_enable: true
130
+ joint_weights:
131
+ ts: 1.0
132
+ text: 1.0
133
+ joint_normalize: standard
134
+ segment_classifier:
135
+ enable: false
136
+ segment_len: 128
137
+ n_segments: 3
138
+ seed: 42
139
+ acd_max_lag: 50
140
+ mdd_bins: 32
141
+ use_cache: false
142
+ cache_file: predictions_cache.pkl
143
+ viz:
144
+ enable: true
145
+ k_cases: 20
146
+ max_vars: 8
147
+ ncols: 5
148
+ seed: 42
149
+ dpi: 200
150
+ alpha: 0.25
151
+ figsize_per_subplot:
152
+ - 3.0
153
+ - 2.0
154
+ output_type: pdf
155
+ output_dir: ''
156
+ created_at: '2025-12-28T16:59:44.641212'
experiments/diffusets/synth-u/seed2/results/expected_seed_metrics.json ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dtw": 5.604377370645617,
3
+ "wape": 89.19720450918459,
4
+ "ed": 9.010427093505859,
5
+ "crps": 0.4444298378533488,
6
+ "acd": 0.06584841758012772,
7
+ "sd": 0.0934342422658183,
8
+ "kd": 0.3979876287319106,
9
+ "mdd": 0.017146341959635412,
10
+ "fid": 39.433125114674,
11
+ "sfd": 9.23097797963554,
12
+ "jftsd": 43.144632222561086,
13
+ "prdc_f1.f1": 0.275845928439235,
14
+ "prdc_f1.precision": 0.63425,
15
+ "prdc_f1.recall": 0.17625,
16
+ "joint_prdc_f1.f1": 0.805401887654133,
17
+ "joint_prdc_f1.precision": 0.93475,
18
+ "joint_prdc_f1.recall": 0.7075,
19
+ "cttp": 26.66297784423828,
20
+ "disc_auc": 0.9999381249999999
21
+ }
experiments/diffusets/synth-u/seed2/summary.json ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ {
2
+ "status": "completed",
3
+ "finished_at": "2025-12-28T17:53:54.368325",
4
+ "best_checkpoint": "checkpoints/finetune/best.ckpt"
5
+ }
experiments/t2s/synth-m/seed0/config.template.yaml ADDED
@@ -0,0 +1,153 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: synth-m_t2s
2
+ description: ''
3
+ seed: 0
4
+ device: cuda:0
5
+ train:
6
+ stages:
7
+ - name: ae_pretrain
8
+ epochs: 200
9
+ lr: 0.001
10
+ use_condition: false
11
+ freeze_modules: []
12
+ early_stopping_patience: 50
13
+ - name: finetune
14
+ epochs: 500
15
+ lr: 0.001
16
+ use_condition: true
17
+ freeze_modules: []
18
+ load_from_stage: ae_pretrain
19
+ early_stopping_patience: 50
20
+ epochs: 700
21
+ batch_size: 256
22
+ lr: 0.001
23
+ weight_decay: 0.0001
24
+ scheduler: cosine
25
+ scheduler_params: {}
26
+ early_stopping_patience: 50
27
+ gradient_clip_val: 1.0
28
+ accumulate_grad_batches: 1
29
+ val_check_interval: 1.0
30
+ num_workers: 4
31
+ pin_memory: true
32
+ limit_train_batches: 1.0
33
+ limit_val_batches: 1.0
34
+ limit_test_batches: 1.0
35
+ num_sanity_val_steps: 2
36
+ log_grad_norm: true
37
+ log_param_norm: true
38
+ log_norm_every_n_steps: 50
39
+ data:
40
+ name: synth-m
41
+ data_folder: __DATA_ROOT__/datasets/synth-m
42
+ n_var: 2
43
+ seq_length: 128
44
+ normalize: false
45
+ train_split: 0.8
46
+ val_split: 0.1
47
+ model:
48
+ name: t2s
49
+ channels: 64
50
+ layers: 4
51
+ nheads: 4
52
+ dropout: 0.1
53
+ patch_size: 2
54
+ mlp_ratio: 2.0
55
+ flow_steps: 10
56
+ cfg_scale: 9.0
57
+ ae:
58
+ num_hiddens: 128
59
+ num_residual_layers: 2
60
+ num_residual_hiddens: 256
61
+ embedding_dim: 64
62
+ condition:
63
+ text:
64
+ enabled: true
65
+ input_dim: 1024
66
+ embedding_key: cap_emb
67
+ dropout: 0.0
68
+ cfg_scale: 1.0
69
+ attribute:
70
+ enabled: false
71
+ continuous_dim: 0
72
+ discrete_configs: []
73
+ output_dim: 128
74
+ dropout: 0.0
75
+ label:
76
+ enabled: false
77
+ num_classes: 10
78
+ output_dim: 64
79
+ fusion: concat
80
+ condition_dropout: 0.0
81
+ eval:
82
+ n_samples: 10
83
+ metrics:
84
+ - dtw
85
+ - wape
86
+ - ed
87
+ - crps
88
+ - acd
89
+ - sd
90
+ - kd
91
+ - mdd
92
+ - fid
93
+ - sfd
94
+ - jftsd
95
+ - prdc_f1
96
+ - joint_prdc_f1
97
+ - cttp
98
+ - disc_auc
99
+ batch_size: 128
100
+ save_samples: true
101
+ sampler: ddim
102
+ display_interval: 10
103
+ clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-m/model_configs.yaml
104
+ clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-m/clip_model_best.pth
105
+ cache_folder: __CACHE_ROOT__/t2s/synth-m/seed0
106
+ use_longalign: false
107
+ reference_split: train
108
+ disc_auc:
109
+ k_folds: 5
110
+ epochs: 5
111
+ batch_size: 128
112
+ lr: 0.001
113
+ weight_decay: 0.0001
114
+ hidden_channels: 128
115
+ num_layers: 3
116
+ dropout: 0.1
117
+ seed: 42
118
+ num_workers: 0
119
+ prdc:
120
+ k: 5
121
+ max_samples: 5000
122
+ seed: 0
123
+ metric: euclidean
124
+ backend: auto
125
+ knn_chunk_size: 4096
126
+ joint_enable: true
127
+ joint_weights:
128
+ ts: 1.0
129
+ text: 1.0
130
+ joint_normalize: standard
131
+ segment_classifier:
132
+ enable: false
133
+ segment_len: 128
134
+ n_segments: 3
135
+ seed: 42
136
+ acd_max_lag: 50
137
+ mdd_bins: 32
138
+ use_cache: false
139
+ cache_file: predictions_cache.pkl
140
+ viz:
141
+ enable: true
142
+ k_cases: 20
143
+ max_vars: 8
144
+ ncols: 5
145
+ seed: 42
146
+ dpi: 200
147
+ alpha: 0.25
148
+ figsize_per_subplot:
149
+ - 3.0
150
+ - 2.0
151
+ output_type: pdf
152
+ output_dir: ''
153
+ created_at: '2025-12-27T23:18:57.418170'
experiments/t2s/synth-m/seed0/results/expected_seed_metrics.json ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dtw": 30.853883330881942,
3
+ "wape": 231.39017954533955,
4
+ "ed": 36.92706576538086,
5
+ "crps": 1.2909548534012474,
6
+ "acd": 0.08280777186155319,
7
+ "sd": 0.43578305680359264,
8
+ "kd": 1.6891005246901758,
9
+ "mdd": 0.027735107421874995,
10
+ "fid": 122.14499871033982,
11
+ "sfd": 317.9769143235806,
12
+ "jftsd": 131.62202535095628,
13
+ "prdc_f1.f1": 0.0,
14
+ "prdc_f1.precision": 0.3175,
15
+ "prdc_f1.recall": 0.0,
16
+ "joint_prdc_f1.f1": 0.012721153846153844,
17
+ "joint_prdc_f1.precision": 0.11025,
18
+ "joint_prdc_f1.recall": 0.00675,
19
+ "cttp": -2.1494174547195435,
20
+ "disc_auc": 1.0
21
+ }
experiments/t2s/synth-m/seed0/summary.json ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ {
2
+ "status": "completed",
3
+ "finished_at": "2025-12-28T00:07:42.509933",
4
+ "best_checkpoint": "checkpoints/finetune/best.ckpt"
5
+ }
experiments/t2s/synth-m/seed1/config.template.yaml ADDED
@@ -0,0 +1,153 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: synth-m_t2s
2
+ description: ''
3
+ seed: 1
4
+ device: cuda:0
5
+ train:
6
+ stages:
7
+ - name: ae_pretrain
8
+ epochs: 200
9
+ lr: 0.001
10
+ use_condition: false
11
+ freeze_modules: []
12
+ early_stopping_patience: 50
13
+ - name: finetune
14
+ epochs: 500
15
+ lr: 0.001
16
+ use_condition: true
17
+ freeze_modules: []
18
+ load_from_stage: ae_pretrain
19
+ early_stopping_patience: 50
20
+ epochs: 700
21
+ batch_size: 256
22
+ lr: 0.001
23
+ weight_decay: 0.0001
24
+ scheduler: cosine
25
+ scheduler_params: {}
26
+ early_stopping_patience: 50
27
+ gradient_clip_val: 1.0
28
+ accumulate_grad_batches: 1
29
+ val_check_interval: 1.0
30
+ num_workers: 4
31
+ pin_memory: true
32
+ limit_train_batches: 1.0
33
+ limit_val_batches: 1.0
34
+ limit_test_batches: 1.0
35
+ num_sanity_val_steps: 2
36
+ log_grad_norm: true
37
+ log_param_norm: true
38
+ log_norm_every_n_steps: 50
39
+ data:
40
+ name: synth-m
41
+ data_folder: __DATA_ROOT__/datasets/synth-m
42
+ n_var: 2
43
+ seq_length: 128
44
+ normalize: false
45
+ train_split: 0.8
46
+ val_split: 0.1
47
+ model:
48
+ name: t2s
49
+ channels: 64
50
+ layers: 4
51
+ nheads: 4
52
+ dropout: 0.1
53
+ patch_size: 2
54
+ mlp_ratio: 2.0
55
+ flow_steps: 10
56
+ cfg_scale: 9.0
57
+ ae:
58
+ num_hiddens: 128
59
+ num_residual_layers: 2
60
+ num_residual_hiddens: 256
61
+ embedding_dim: 64
62
+ condition:
63
+ text:
64
+ enabled: true
65
+ input_dim: 1024
66
+ embedding_key: cap_emb
67
+ dropout: 0.0
68
+ cfg_scale: 1.0
69
+ attribute:
70
+ enabled: false
71
+ continuous_dim: 0
72
+ discrete_configs: []
73
+ output_dim: 128
74
+ dropout: 0.0
75
+ label:
76
+ enabled: false
77
+ num_classes: 10
78
+ output_dim: 64
79
+ fusion: concat
80
+ condition_dropout: 0.0
81
+ eval:
82
+ n_samples: 10
83
+ metrics:
84
+ - dtw
85
+ - wape
86
+ - ed
87
+ - crps
88
+ - acd
89
+ - sd
90
+ - kd
91
+ - mdd
92
+ - fid
93
+ - sfd
94
+ - jftsd
95
+ - prdc_f1
96
+ - joint_prdc_f1
97
+ - cttp
98
+ - disc_auc
99
+ batch_size: 128
100
+ save_samples: true
101
+ sampler: ddim
102
+ display_interval: 10
103
+ clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-m/model_configs.yaml
104
+ clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-m/clip_model_best.pth
105
+ cache_folder: __CACHE_ROOT__/t2s/synth-m/seed1
106
+ use_longalign: false
107
+ reference_split: train
108
+ disc_auc:
109
+ k_folds: 5
110
+ epochs: 5
111
+ batch_size: 128
112
+ lr: 0.001
113
+ weight_decay: 0.0001
114
+ hidden_channels: 128
115
+ num_layers: 3
116
+ dropout: 0.1
117
+ seed: 42
118
+ num_workers: 0
119
+ prdc:
120
+ k: 5
121
+ max_samples: 5000
122
+ seed: 0
123
+ metric: euclidean
124
+ backend: auto
125
+ knn_chunk_size: 4096
126
+ joint_enable: true
127
+ joint_weights:
128
+ ts: 1.0
129
+ text: 1.0
130
+ joint_normalize: standard
131
+ segment_classifier:
132
+ enable: false
133
+ segment_len: 128
134
+ n_segments: 3
135
+ seed: 42
136
+ acd_max_lag: 50
137
+ mdd_bins: 32
138
+ use_cache: false
139
+ cache_file: predictions_cache.pkl
140
+ viz:
141
+ enable: true
142
+ k_cases: 20
143
+ max_vars: 8
144
+ ncols: 5
145
+ seed: 42
146
+ dpi: 200
147
+ alpha: 0.25
148
+ figsize_per_subplot:
149
+ - 3.0
150
+ - 2.0
151
+ output_type: pdf
152
+ output_dir: ''
153
+ created_at: '2025-12-27T23:29:31.014346'
experiments/t2s/synth-m/seed1/results/expected_seed_metrics.json ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dtw": 51.97795645532827,
3
+ "wape": 375.2316361826495,
4
+ "ed": 57.725555297851564,
5
+ "crps": 2.47075430506335,
6
+ "acd": 0.036171168088912964,
7
+ "sd": 0.48953894824419564,
8
+ "kd": 1.4351366797985663,
9
+ "mdd": 0.03595974731445312,
10
+ "fid": 109.6813469453873,
11
+ "sfd": 409.9287758892757,
12
+ "jftsd": 118.27890977769226,
13
+ "prdc_f1.f1": 0.0,
14
+ "prdc_f1.precision": 0.4685,
15
+ "prdc_f1.recall": 0.0,
16
+ "joint_prdc_f1.f1": 0.02031169665809768,
17
+ "joint_prdc_f1.precision": 0.18375,
18
+ "joint_prdc_f1.recall": 0.01075,
19
+ "cttp": 4.7206211853027344,
20
+ "disc_auc": 1.0
21
+ }
experiments/t2s/synth-m/seed1/summary.json ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ {
2
+ "status": "completed",
3
+ "finished_at": "2025-12-28T00:09:22.363900",
4
+ "best_checkpoint": "checkpoints/finetune/best.ckpt"
5
+ }
experiments/t2s/synth-m/seed2/config.template.yaml ADDED
@@ -0,0 +1,153 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: synth-m_t2s
2
+ description: ''
3
+ seed: 2
4
+ device: cuda:0
5
+ train:
6
+ stages:
7
+ - name: ae_pretrain
8
+ epochs: 200
9
+ lr: 0.001
10
+ use_condition: false
11
+ freeze_modules: []
12
+ early_stopping_patience: 50
13
+ - name: finetune
14
+ epochs: 500
15
+ lr: 0.001
16
+ use_condition: true
17
+ freeze_modules: []
18
+ load_from_stage: ae_pretrain
19
+ early_stopping_patience: 50
20
+ epochs: 700
21
+ batch_size: 256
22
+ lr: 0.001
23
+ weight_decay: 0.0001
24
+ scheduler: cosine
25
+ scheduler_params: {}
26
+ early_stopping_patience: 50
27
+ gradient_clip_val: 1.0
28
+ accumulate_grad_batches: 1
29
+ val_check_interval: 1.0
30
+ num_workers: 4
31
+ pin_memory: true
32
+ limit_train_batches: 1.0
33
+ limit_val_batches: 1.0
34
+ limit_test_batches: 1.0
35
+ num_sanity_val_steps: 2
36
+ log_grad_norm: true
37
+ log_param_norm: true
38
+ log_norm_every_n_steps: 50
39
+ data:
40
+ name: synth-m
41
+ data_folder: __DATA_ROOT__/datasets/synth-m
42
+ n_var: 2
43
+ seq_length: 128
44
+ normalize: false
45
+ train_split: 0.8
46
+ val_split: 0.1
47
+ model:
48
+ name: t2s
49
+ channels: 64
50
+ layers: 4
51
+ nheads: 4
52
+ dropout: 0.1
53
+ patch_size: 2
54
+ mlp_ratio: 2.0
55
+ flow_steps: 10
56
+ cfg_scale: 9.0
57
+ ae:
58
+ num_hiddens: 128
59
+ num_residual_layers: 2
60
+ num_residual_hiddens: 256
61
+ embedding_dim: 64
62
+ condition:
63
+ text:
64
+ enabled: true
65
+ input_dim: 1024
66
+ embedding_key: cap_emb
67
+ dropout: 0.0
68
+ cfg_scale: 1.0
69
+ attribute:
70
+ enabled: false
71
+ continuous_dim: 0
72
+ discrete_configs: []
73
+ output_dim: 128
74
+ dropout: 0.0
75
+ label:
76
+ enabled: false
77
+ num_classes: 10
78
+ output_dim: 64
79
+ fusion: concat
80
+ condition_dropout: 0.0
81
+ eval:
82
+ n_samples: 10
83
+ metrics:
84
+ - dtw
85
+ - wape
86
+ - ed
87
+ - crps
88
+ - acd
89
+ - sd
90
+ - kd
91
+ - mdd
92
+ - fid
93
+ - sfd
94
+ - jftsd
95
+ - prdc_f1
96
+ - joint_prdc_f1
97
+ - cttp
98
+ - disc_auc
99
+ batch_size: 128
100
+ save_samples: true
101
+ sampler: ddim
102
+ display_interval: 10
103
+ clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-m/model_configs.yaml
104
+ clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-m/clip_model_best.pth
105
+ cache_folder: __CACHE_ROOT__/t2s/synth-m/seed2
106
+ use_longalign: false
107
+ reference_split: train
108
+ disc_auc:
109
+ k_folds: 5
110
+ epochs: 5
111
+ batch_size: 128
112
+ lr: 0.001
113
+ weight_decay: 0.0001
114
+ hidden_channels: 128
115
+ num_layers: 3
116
+ dropout: 0.1
117
+ seed: 42
118
+ num_workers: 0
119
+ prdc:
120
+ k: 5
121
+ max_samples: 5000
122
+ seed: 0
123
+ metric: euclidean
124
+ backend: auto
125
+ knn_chunk_size: 4096
126
+ joint_enable: true
127
+ joint_weights:
128
+ ts: 1.0
129
+ text: 1.0
130
+ joint_normalize: standard
131
+ segment_classifier:
132
+ enable: false
133
+ segment_len: 128
134
+ n_segments: 3
135
+ seed: 42
136
+ acd_max_lag: 50
137
+ mdd_bins: 32
138
+ use_cache: false
139
+ cache_file: predictions_cache.pkl
140
+ viz:
141
+ enable: true
142
+ k_cases: 20
143
+ max_vars: 8
144
+ ncols: 5
145
+ seed: 42
146
+ dpi: 200
147
+ alpha: 0.25
148
+ figsize_per_subplot:
149
+ - 3.0
150
+ - 2.0
151
+ output_type: pdf
152
+ output_dir: ''
153
+ created_at: '2025-12-27T23:36:31.656409'
experiments/t2s/synth-m/seed2/results/expected_seed_metrics.json ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dtw": 33.4548552772021,
3
+ "wape": 233.52481543411972,
4
+ "ed": 38.108337890625,
5
+ "crps": 1.2779978524073943,
6
+ "acd": 0.10699678212404251,
7
+ "sd": 0.26627323448919915,
8
+ "kd": 1.3601897974690218,
9
+ "mdd": 0.027756306966145835,
10
+ "fid": 109.19084633158991,
11
+ "sfd": 384.584061896183,
12
+ "jftsd": 119.15697446699316,
13
+ "prdc_f1.f1": 0.0,
14
+ "prdc_f1.precision": 0.48775,
15
+ "prdc_f1.recall": 0.0,
16
+ "joint_prdc_f1.f1": 0.02104734576757532,
17
+ "joint_prdc_f1.precision": 0.163,
18
+ "joint_prdc_f1.recall": 0.01125,
19
+ "cttp": 2.3761753463745117,
20
+ "disc_auc": 1.0
21
+ }
experiments/t2s/synth-m/seed2/summary.json ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ {
2
+ "status": "completed",
3
+ "finished_at": "2025-12-28T00:23:21.692223",
4
+ "best_checkpoint": "checkpoints/finetune/best.ckpt"
5
+ }
experiments/t2s/synth-u/seed0/config.template.yaml ADDED
@@ -0,0 +1,153 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: synth-u_t2s
2
+ description: ''
3
+ seed: 0
4
+ device: cuda:0
5
+ train:
6
+ stages:
7
+ - name: ae_pretrain
8
+ epochs: 200
9
+ lr: 0.001
10
+ use_condition: false
11
+ freeze_modules: []
12
+ early_stopping_patience: 50
13
+ - name: finetune
14
+ epochs: 500
15
+ lr: 0.001
16
+ use_condition: true
17
+ freeze_modules: []
18
+ load_from_stage: ae_pretrain
19
+ early_stopping_patience: 50
20
+ epochs: 700
21
+ batch_size: 256
22
+ lr: 0.001
23
+ weight_decay: 0.0001
24
+ scheduler: cosine
25
+ scheduler_params: {}
26
+ early_stopping_patience: 50
27
+ gradient_clip_val: 1.0
28
+ accumulate_grad_batches: 1
29
+ val_check_interval: 1.0
30
+ num_workers: 4
31
+ pin_memory: true
32
+ limit_train_batches: 1.0
33
+ limit_val_batches: 1.0
34
+ limit_test_batches: 1.0
35
+ num_sanity_val_steps: 2
36
+ log_grad_norm: true
37
+ log_param_norm: true
38
+ log_norm_every_n_steps: 50
39
+ data:
40
+ name: synth-u
41
+ data_folder: __DATA_ROOT__/datasets/synth-u
42
+ n_var: 1
43
+ seq_length: 128
44
+ normalize: false
45
+ train_split: 0.8
46
+ val_split: 0.1
47
+ model:
48
+ name: t2s
49
+ channels: 64
50
+ layers: 4
51
+ nheads: 4
52
+ dropout: 0.1
53
+ patch_size: 2
54
+ mlp_ratio: 2.0
55
+ flow_steps: 10
56
+ cfg_scale: 9.0
57
+ ae:
58
+ num_hiddens: 128
59
+ num_residual_layers: 2
60
+ num_residual_hiddens: 256
61
+ embedding_dim: 64
62
+ condition:
63
+ text:
64
+ enabled: true
65
+ input_dim: 1024
66
+ embedding_key: cap_emb
67
+ dropout: 0.0
68
+ cfg_scale: 1.0
69
+ attribute:
70
+ enabled: false
71
+ continuous_dim: 0
72
+ discrete_configs: []
73
+ output_dim: 128
74
+ dropout: 0.0
75
+ label:
76
+ enabled: false
77
+ num_classes: 10
78
+ output_dim: 64
79
+ fusion: concat
80
+ condition_dropout: 0.0
81
+ eval:
82
+ n_samples: 10
83
+ metrics:
84
+ - dtw
85
+ - wape
86
+ - ed
87
+ - crps
88
+ - acd
89
+ - sd
90
+ - kd
91
+ - mdd
92
+ - fid
93
+ - sfd
94
+ - jftsd
95
+ - prdc_f1
96
+ - joint_prdc_f1
97
+ - cttp
98
+ - disc_auc
99
+ batch_size: 128
100
+ save_samples: true
101
+ sampler: ddim
102
+ display_interval: 10
103
+ clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-u/model_configs.yaml
104
+ clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-u/clip_model_best.pth
105
+ cache_folder: __CACHE_ROOT__/t2s/synth-u/seed0
106
+ use_longalign: false
107
+ reference_split: train
108
+ disc_auc:
109
+ k_folds: 5
110
+ epochs: 5
111
+ batch_size: 128
112
+ lr: 0.001
113
+ weight_decay: 0.0001
114
+ hidden_channels: 128
115
+ num_layers: 3
116
+ dropout: 0.1
117
+ seed: 42
118
+ num_workers: 0
119
+ prdc:
120
+ k: 5
121
+ max_samples: 5000
122
+ seed: 0
123
+ metric: euclidean
124
+ backend: auto
125
+ knn_chunk_size: 4096
126
+ joint_enable: true
127
+ joint_weights:
128
+ ts: 1.0
129
+ text: 1.0
130
+ joint_normalize: standard
131
+ segment_classifier:
132
+ enable: false
133
+ segment_len: 128
134
+ n_segments: 3
135
+ seed: 42
136
+ acd_max_lag: 50
137
+ mdd_bins: 32
138
+ use_cache: false
139
+ cache_file: predictions_cache.pkl
140
+ viz:
141
+ enable: true
142
+ k_cases: 20
143
+ max_vars: 8
144
+ ncols: 5
145
+ seed: 42
146
+ dpi: 200
147
+ alpha: 0.25
148
+ figsize_per_subplot:
149
+ - 3.0
150
+ - 2.0
151
+ output_type: pdf
152
+ output_dir: ''
153
+ created_at: '2025-12-27T23:40:52.259276'
experiments/t2s/synth-u/seed0/results/expected_seed_metrics.json ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dtw": 47.12652707117338,
3
+ "wape": 550.2959810862212,
4
+ "ed": 54.415912963867186,
5
+ "crps": 3.4977831919962763,
6
+ "acd": 0.039769306778907776,
7
+ "sd": 0.32305185133674863,
8
+ "kd": 0.5842860290533332,
9
+ "mdd": 0.05455151367187498,
10
+ "fid": 175.4499695138689,
11
+ "sfd": 158.77707966819406,
12
+ "jftsd": 186.9348911336512,
13
+ "prdc_f1.f1": 0.0,
14
+ "prdc_f1.precision": 0.01575,
15
+ "prdc_f1.recall": 0.0,
16
+ "joint_prdc_f1.f1": 0.005913978494623656,
17
+ "joint_prdc_f1.precision": 0.20625,
18
+ "joint_prdc_f1.recall": 0.003,
19
+ "cttp": 5.37317338180542,
20
+ "disc_auc": 1.0
21
+ }
experiments/t2s/synth-u/seed0/summary.json ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ {
2
+ "status": "completed",
3
+ "finished_at": "2025-12-28T00:15:00.860457",
4
+ "best_checkpoint": "checkpoints/finetune/best.ckpt"
5
+ }
experiments/t2s/synth-u/seed1/config.template.yaml ADDED
@@ -0,0 +1,153 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: synth-u_t2s
2
+ description: ''
3
+ seed: 1
4
+ device: cuda:0
5
+ train:
6
+ stages:
7
+ - name: ae_pretrain
8
+ epochs: 200
9
+ lr: 0.001
10
+ use_condition: false
11
+ freeze_modules: []
12
+ early_stopping_patience: 50
13
+ - name: finetune
14
+ epochs: 500
15
+ lr: 0.001
16
+ use_condition: true
17
+ freeze_modules: []
18
+ load_from_stage: ae_pretrain
19
+ early_stopping_patience: 50
20
+ epochs: 700
21
+ batch_size: 256
22
+ lr: 0.001
23
+ weight_decay: 0.0001
24
+ scheduler: cosine
25
+ scheduler_params: {}
26
+ early_stopping_patience: 50
27
+ gradient_clip_val: 1.0
28
+ accumulate_grad_batches: 1
29
+ val_check_interval: 1.0
30
+ num_workers: 4
31
+ pin_memory: true
32
+ limit_train_batches: 1.0
33
+ limit_val_batches: 1.0
34
+ limit_test_batches: 1.0
35
+ num_sanity_val_steps: 2
36
+ log_grad_norm: true
37
+ log_param_norm: true
38
+ log_norm_every_n_steps: 50
39
+ data:
40
+ name: synth-u
41
+ data_folder: __DATA_ROOT__/datasets/synth-u
42
+ n_var: 1
43
+ seq_length: 128
44
+ normalize: false
45
+ train_split: 0.8
46
+ val_split: 0.1
47
+ model:
48
+ name: t2s
49
+ channels: 64
50
+ layers: 4
51
+ nheads: 4
52
+ dropout: 0.1
53
+ patch_size: 2
54
+ mlp_ratio: 2.0
55
+ flow_steps: 10
56
+ cfg_scale: 9.0
57
+ ae:
58
+ num_hiddens: 128
59
+ num_residual_layers: 2
60
+ num_residual_hiddens: 256
61
+ embedding_dim: 64
62
+ condition:
63
+ text:
64
+ enabled: true
65
+ input_dim: 1024
66
+ embedding_key: cap_emb
67
+ dropout: 0.0
68
+ cfg_scale: 1.0
69
+ attribute:
70
+ enabled: false
71
+ continuous_dim: 0
72
+ discrete_configs: []
73
+ output_dim: 128
74
+ dropout: 0.0
75
+ label:
76
+ enabled: false
77
+ num_classes: 10
78
+ output_dim: 64
79
+ fusion: concat
80
+ condition_dropout: 0.0
81
+ eval:
82
+ n_samples: 10
83
+ metrics:
84
+ - dtw
85
+ - wape
86
+ - ed
87
+ - crps
88
+ - acd
89
+ - sd
90
+ - kd
91
+ - mdd
92
+ - fid
93
+ - sfd
94
+ - jftsd
95
+ - prdc_f1
96
+ - joint_prdc_f1
97
+ - cttp
98
+ - disc_auc
99
+ batch_size: 128
100
+ save_samples: true
101
+ sampler: ddim
102
+ display_interval: 10
103
+ clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-u/model_configs.yaml
104
+ clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-u/clip_model_best.pth
105
+ cache_folder: __CACHE_ROOT__/t2s/synth-u/seed1
106
+ use_longalign: false
107
+ reference_split: train
108
+ disc_auc:
109
+ k_folds: 5
110
+ epochs: 5
111
+ batch_size: 128
112
+ lr: 0.001
113
+ weight_decay: 0.0001
114
+ hidden_channels: 128
115
+ num_layers: 3
116
+ dropout: 0.1
117
+ seed: 42
118
+ num_workers: 0
119
+ prdc:
120
+ k: 5
121
+ max_samples: 5000
122
+ seed: 0
123
+ metric: euclidean
124
+ backend: auto
125
+ knn_chunk_size: 4096
126
+ joint_enable: true
127
+ joint_weights:
128
+ ts: 1.0
129
+ text: 1.0
130
+ joint_normalize: standard
131
+ segment_classifier:
132
+ enable: false
133
+ segment_len: 128
134
+ n_segments: 3
135
+ seed: 42
136
+ acd_max_lag: 50
137
+ mdd_bins: 32
138
+ use_cache: false
139
+ cache_file: predictions_cache.pkl
140
+ viz:
141
+ enable: true
142
+ k_cases: 20
143
+ max_vars: 8
144
+ ncols: 5
145
+ seed: 42
146
+ dpi: 200
147
+ alpha: 0.25
148
+ figsize_per_subplot:
149
+ - 3.0
150
+ - 2.0
151
+ output_type: pdf
152
+ output_dir: ''
153
+ created_at: '2025-12-27T23:41:04.392813'
experiments/t2s/synth-u/seed1/results/expected_seed_metrics.json ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dtw": 15.365408773958965,
3
+ "wape": 178.296819353186,
4
+ "ed": 20.013489288330078,
5
+ "crps": 1.0322840045545667,
6
+ "acd": 0.04774828627705574,
7
+ "sd": 0.9647802643913168,
8
+ "kd": 1.2767695148803266,
9
+ "mdd": 0.02350620524088541,
10
+ "fid": 134.2823978837169,
11
+ "sfd": 79.6028833155586,
12
+ "jftsd": 147.3762497463016,
13
+ "prdc_f1.f1": 0.0,
14
+ "prdc_f1.precision": 0.4055,
15
+ "prdc_f1.recall": 0.0,
16
+ "joint_prdc_f1.f1": 0.027200400000000003,
17
+ "joint_prdc_f1.precision": 0.29825,
18
+ "joint_prdc_f1.recall": 0.01425,
19
+ "cttp": 12.133055778503419,
20
+ "disc_auc": 1.0
21
+ }