Add files using upload-large-folder tool
Browse filesThis view is limited to 50 files because it contains too many changes.
See raw diff
- experiments/bridge/synth-m/seed0/config.template.yaml +169 -0
- experiments/bridge/synth-m/seed0/results/expected_seed_metrics.json +21 -0
- experiments/bridge/synth-m/seed0/summary.json +5 -0
- experiments/bridge/synth-m/seed1/config.template.yaml +169 -0
- experiments/bridge/synth-m/seed1/results/expected_seed_metrics.json +21 -0
- experiments/bridge/synth-m/seed1/summary.json +5 -0
- experiments/bridge/synth-m/seed2/config.template.yaml +169 -0
- experiments/bridge/synth-m/seed2/results/expected_seed_metrics.json +21 -0
- experiments/bridge/synth-m/seed2/summary.json +5 -0
- experiments/bridge/synth-u/seed0/config.template.yaml +169 -0
- experiments/bridge/synth-u/seed0/results/expected_seed_metrics.json +21 -0
- experiments/bridge/synth-u/seed0/summary.json +5 -0
- experiments/bridge/synth-u/seed1/config.template.yaml +169 -0
- experiments/bridge/synth-u/seed1/results/expected_seed_metrics.json +21 -0
- experiments/bridge/synth-u/seed1/summary.json +5 -0
- experiments/bridge/synth-u/seed2/config.template.yaml +169 -0
- experiments/bridge/synth-u/seed2/results/expected_seed_metrics.json +21 -0
- experiments/bridge/synth-u/seed2/summary.json +5 -0
- experiments/diffusets/synth-m/seed0/config.template.yaml +156 -0
- experiments/diffusets/synth-m/seed0/results/expected_seed_metrics.json +21 -0
- experiments/diffusets/synth-m/seed0/summary.json +5 -0
- experiments/diffusets/synth-m/seed1/config.template.yaml +156 -0
- experiments/diffusets/synth-m/seed1/results/expected_seed_metrics.json +21 -0
- experiments/diffusets/synth-m/seed1/summary.json +5 -0
- experiments/diffusets/synth-m/seed2/config.template.yaml +156 -0
- experiments/diffusets/synth-m/seed2/results/expected_seed_metrics.json +21 -0
- experiments/diffusets/synth-m/seed2/summary.json +5 -0
- experiments/diffusets/synth-u/seed0/config.template.yaml +156 -0
- experiments/diffusets/synth-u/seed0/results/expected_seed_metrics.json +21 -0
- experiments/diffusets/synth-u/seed0/summary.json +5 -0
- experiments/diffusets/synth-u/seed1/config.template.yaml +156 -0
- experiments/diffusets/synth-u/seed1/results/expected_seed_metrics.json +21 -0
- experiments/diffusets/synth-u/seed1/summary.json +5 -0
- experiments/diffusets/synth-u/seed2/config.template.yaml +156 -0
- experiments/diffusets/synth-u/seed2/results/expected_seed_metrics.json +21 -0
- experiments/diffusets/synth-u/seed2/summary.json +5 -0
- experiments/t2s/synth-m/seed0/config.template.yaml +153 -0
- experiments/t2s/synth-m/seed0/results/expected_seed_metrics.json +21 -0
- experiments/t2s/synth-m/seed0/summary.json +5 -0
- experiments/t2s/synth-m/seed1/config.template.yaml +153 -0
- experiments/t2s/synth-m/seed1/results/expected_seed_metrics.json +21 -0
- experiments/t2s/synth-m/seed1/summary.json +5 -0
- experiments/t2s/synth-m/seed2/config.template.yaml +153 -0
- experiments/t2s/synth-m/seed2/results/expected_seed_metrics.json +21 -0
- experiments/t2s/synth-m/seed2/summary.json +5 -0
- experiments/t2s/synth-u/seed0/config.template.yaml +153 -0
- experiments/t2s/synth-u/seed0/results/expected_seed_metrics.json +21 -0
- experiments/t2s/synth-u/seed0/summary.json +5 -0
- experiments/t2s/synth-u/seed1/config.template.yaml +153 -0
- 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 |
+
}
|