diff --git a/experiments/bridge/synth-m/seed0/config.template.yaml b/experiments/bridge/synth-m/seed0/config.template.yaml new file mode 100644 index 0000000000000000000000000000000000000000..c05e72503c4fbbc90a2368082b97fb8a8045a660 --- /dev/null +++ b/experiments/bridge/synth-m/seed0/config.template.yaml @@ -0,0 +1,169 @@ +name: synth-m_bridge +description: '' +seed: 0 +device: cuda:0 +train: + stages: + - name: finetune + epochs: 700 + lr: 0.001 + use_condition: true + freeze_modules: [] + early_stopping_patience: 50 + epochs: 700 + batch_size: 256 + lr: 0.001 + weight_decay: 0.0001 + scheduler: cosine + scheduler_params: {} + early_stopping_patience: 50 + gradient_clip_val: 1.0 + accumulate_grad_batches: 1 + val_check_interval: 1.0 + num_workers: 4 + pin_memory: true + limit_train_batches: 1.0 + limit_val_batches: 1.0 + limit_test_batches: 1.0 + num_sanity_val_steps: 2 + log_grad_norm: true + log_param_norm: true + log_norm_every_n_steps: 50 +data: + name: synth-m + data_folder: __DATA_ROOT__/datasets/synth-m + n_var: 2 + seq_length: 128 + normalize: false + train_split: 0.8 + val_split: 0.1 +model: + name: bridge + channels: 64 + layers: 3 + nheads: 8 + dropout: 0.0 + latent_dim: 32 + num_latents: 16 + latent_unit: 1 + repre_emb_channels: 32 + prototype_dim: 32 + context_dim: 32 + text_dim: 1024 + fusion_type: gated_add + cond_drop_prob: 0.5 + use_cfg: true + model_channels: 64 + num_res_blocks: 2 + attention_resolutions: + - 1 + - 2 + - 4 + channel_mult: + - 1 + - 2 + - 4 + - 4 + num_heads: 8 + num_head_channels: -1 + use_scale_shift_norm: true + resblock_updown: true + use_spatial_transformer: true + transformer_depth: 1 + diffusion_steps: 50 + noise_schedule: quad + beta_start: 0.0001 + beta_end: 0.5 + bridge_type: vp +condition: + text: + enabled: true + input_dim: 1024 + embedding_key: cap_emb + dropout: 0.0 + cfg_scale: 1.0 + attribute: + enabled: false + continuous_dim: 0 + discrete_configs: [] + output_dim: 128 + dropout: 0.0 + label: + enabled: false + num_classes: 10 + output_dim: 64 + fusion: concat + condition_dropout: 0.0 +eval: + n_samples: 10 + metrics: + - dtw + - wape + - ed + - crps + - acd + - sd + - kd + - mdd + - fid + - sfd + - jftsd + - prdc_f1 + - joint_prdc_f1 + - cttp + - disc_auc + batch_size: 128 + save_samples: true + sampler: ddim + display_interval: 10 + clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-m/model_configs.yaml + clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-m/clip_model_best.pth + cache_folder: __CACHE_ROOT__/bridge/synth-m/seed0 + use_longalign: false + reference_split: train + disc_auc: + k_folds: 5 + epochs: 5 + batch_size: 128 + lr: 0.001 + weight_decay: 0.0001 + hidden_channels: 128 + num_layers: 3 + dropout: 0.1 + seed: 42 + num_workers: 0 + prdc: + k: 5 + max_samples: 5000 + seed: 0 + metric: euclidean + backend: auto + knn_chunk_size: 4096 + joint_enable: true + joint_weights: + ts: 1.0 + text: 1.0 + joint_normalize: standard + segment_classifier: + enable: false + segment_len: 128 + n_segments: 3 + seed: 42 + acd_max_lag: 50 + mdd_bins: 32 + use_cache: false + cache_file: predictions_cache.pkl + viz: + enable: true + k_cases: 20 + max_vars: 8 + ncols: 5 + seed: 42 + dpi: 200 + alpha: 0.25 + figsize_per_subplot: + - 3.0 + - 2.0 + output_type: pdf +output_dir: '' +created_at: '2025-12-28T18:30:19.791690' diff --git a/experiments/bridge/synth-m/seed0/results/expected_seed_metrics.json b/experiments/bridge/synth-m/seed0/results/expected_seed_metrics.json new file mode 100644 index 0000000000000000000000000000000000000000..051b03c81948d4429bf22fc5ded1ee5b8bb2d2f7 --- /dev/null +++ b/experiments/bridge/synth-m/seed0/results/expected_seed_metrics.json @@ -0,0 +1,21 @@ +{ + "dtw": 13.817281705569064, + "wape": 106.19159049917226, + "ed": 16.047494094848634, + "crps": 0.6222953609760046, + "acd": 0.04810057580471039, + "sd": 0.3342529663983614, + "kd": 0.2772295471154491, + "mdd": 0.029095235188802075, + "fid": 36.90573498644184, + "sfd": 46.77665048709039, + "jftsd": 44.922412953420775, + "prdc_f1.f1": 0.2880602782071098, + "prdc_f1.precision": 0.62125, + "prdc_f1.recall": 0.1875, + "joint_prdc_f1.f1": 0.33039791744142805, + "joint_prdc_f1.precision": 0.38, + "joint_prdc_f1.recall": 0.29225, + "cttp": 18.433125, + "disc_auc": 0.9997595312499999 +} \ No newline at end of file diff --git a/experiments/bridge/synth-m/seed0/summary.json b/experiments/bridge/synth-m/seed0/summary.json new file mode 100644 index 0000000000000000000000000000000000000000..b7c859bc0ef318bc5aa71b63b0f8cf679d3229c1 --- /dev/null +++ b/experiments/bridge/synth-m/seed0/summary.json @@ -0,0 +1,5 @@ +{ + "status": "completed", + "finished_at": "2025-12-28T22:05:24.380146", + "best_checkpoint": "checkpoints/finetune/best.ckpt" +} \ No newline at end of file diff --git a/experiments/bridge/synth-m/seed1/config.template.yaml b/experiments/bridge/synth-m/seed1/config.template.yaml new file mode 100644 index 0000000000000000000000000000000000000000..a4d2643a59e634bffeddf5a1c0a59524eb3c2040 --- /dev/null +++ b/experiments/bridge/synth-m/seed1/config.template.yaml @@ -0,0 +1,169 @@ +name: synth-m_bridge +description: '' +seed: 1 +device: cuda:0 +train: + stages: + - name: finetune + epochs: 700 + lr: 0.001 + use_condition: true + freeze_modules: [] + early_stopping_patience: 50 + epochs: 700 + batch_size: 256 + lr: 0.001 + weight_decay: 0.0001 + scheduler: cosine + scheduler_params: {} + early_stopping_patience: 50 + gradient_clip_val: 1.0 + accumulate_grad_batches: 1 + val_check_interval: 1.0 + num_workers: 4 + pin_memory: true + limit_train_batches: 1.0 + limit_val_batches: 1.0 + limit_test_batches: 1.0 + num_sanity_val_steps: 2 + log_grad_norm: true + log_param_norm: true + log_norm_every_n_steps: 50 +data: + name: synth-m + data_folder: __DATA_ROOT__/datasets/synth-m + n_var: 2 + seq_length: 128 + normalize: false + train_split: 0.8 + val_split: 0.1 +model: + name: bridge + channels: 64 + layers: 3 + nheads: 8 + dropout: 0.0 + latent_dim: 32 + num_latents: 16 + latent_unit: 1 + repre_emb_channels: 32 + prototype_dim: 32 + context_dim: 32 + text_dim: 1024 + fusion_type: gated_add + cond_drop_prob: 0.5 + use_cfg: true + model_channels: 64 + num_res_blocks: 2 + attention_resolutions: + - 1 + - 2 + - 4 + channel_mult: + - 1 + - 2 + - 4 + - 4 + num_heads: 8 + num_head_channels: -1 + use_scale_shift_norm: true + resblock_updown: true + use_spatial_transformer: true + transformer_depth: 1 + diffusion_steps: 50 + noise_schedule: quad + beta_start: 0.0001 + beta_end: 0.5 + bridge_type: vp +condition: + text: + enabled: true + input_dim: 1024 + embedding_key: cap_emb + dropout: 0.0 + cfg_scale: 1.0 + attribute: + enabled: false + continuous_dim: 0 + discrete_configs: [] + output_dim: 128 + dropout: 0.0 + label: + enabled: false + num_classes: 10 + output_dim: 64 + fusion: concat + condition_dropout: 0.0 +eval: + n_samples: 10 + metrics: + - dtw + - wape + - ed + - crps + - acd + - sd + - kd + - mdd + - fid + - sfd + - jftsd + - prdc_f1 + - joint_prdc_f1 + - cttp + - disc_auc + batch_size: 128 + save_samples: true + sampler: ddim + display_interval: 10 + clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-m/model_configs.yaml + clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-m/clip_model_best.pth + cache_folder: __CACHE_ROOT__/bridge/synth-m/seed1 + use_longalign: false + reference_split: train + disc_auc: + k_folds: 5 + epochs: 5 + batch_size: 128 + lr: 0.001 + weight_decay: 0.0001 + hidden_channels: 128 + num_layers: 3 + dropout: 0.1 + seed: 42 + num_workers: 0 + prdc: + k: 5 + max_samples: 5000 + seed: 0 + metric: euclidean + backend: auto + knn_chunk_size: 4096 + joint_enable: true + joint_weights: + ts: 1.0 + text: 1.0 + joint_normalize: standard + segment_classifier: + enable: false + segment_len: 128 + n_segments: 3 + seed: 42 + acd_max_lag: 50 + mdd_bins: 32 + use_cache: false + cache_file: predictions_cache.pkl + viz: + enable: true + k_cases: 20 + max_vars: 8 + ncols: 5 + seed: 42 + dpi: 200 + alpha: 0.25 + figsize_per_subplot: + - 3.0 + - 2.0 + output_type: pdf +output_dir: '' +created_at: '2025-12-28T18:32:05.306146' diff --git a/experiments/bridge/synth-m/seed1/results/expected_seed_metrics.json b/experiments/bridge/synth-m/seed1/results/expected_seed_metrics.json new file mode 100644 index 0000000000000000000000000000000000000000..41f5ddff4bf6d69fe837d8fcb4d3d64b510b4738 --- /dev/null +++ b/experiments/bridge/synth-m/seed1/results/expected_seed_metrics.json @@ -0,0 +1,21 @@ +{ + "dtw": 11.941985574008141, + "wape": 97.19576751179419, + "ed": 14.614929718017578, + "crps": 0.5933726656309234, + "acd": 0.08524112403392792, + "sd": 0.027004171864705717, + "kd": 0.707300600732306, + "mdd": 0.02950222778320312, + "fid": 60.61234598914979, + "sfd": 45.196805251014396, + "jftsd": 68.15098610695699, + "prdc_f1.f1": 0.17119105807935597, + "prdc_f1.precision": 0.77325, + "prdc_f1.recall": 0.09625, + "joint_prdc_f1.f1": 0.19173251468232783, + "joint_prdc_f1.precision": 0.33375, + "joint_prdc_f1.recall": 0.1345, + "cttp": 18.584201385498048, + "disc_auc": 0.9996928125 +} \ No newline at end of file diff --git a/experiments/bridge/synth-m/seed1/summary.json b/experiments/bridge/synth-m/seed1/summary.json new file mode 100644 index 0000000000000000000000000000000000000000..28d7c70b9b052ba70bd4ec840e34b86cc3dd210c --- /dev/null +++ b/experiments/bridge/synth-m/seed1/summary.json @@ -0,0 +1,5 @@ +{ + "status": "completed", + "finished_at": "2025-12-28T19:25:34.577642", + "best_checkpoint": "checkpoints/finetune/best.ckpt" +} \ No newline at end of file diff --git a/experiments/bridge/synth-m/seed2/config.template.yaml b/experiments/bridge/synth-m/seed2/config.template.yaml new file mode 100644 index 0000000000000000000000000000000000000000..13621718c6ebefac9f94f8ca451ebb5fb34cd92e --- /dev/null +++ b/experiments/bridge/synth-m/seed2/config.template.yaml @@ -0,0 +1,169 @@ +name: synth-m_bridge +description: '' +seed: 2 +device: cuda:0 +train: + stages: + - name: finetune + epochs: 700 + lr: 0.001 + use_condition: true + freeze_modules: [] + early_stopping_patience: 50 + epochs: 700 + batch_size: 256 + lr: 0.001 + weight_decay: 0.0001 + scheduler: cosine + scheduler_params: {} + early_stopping_patience: 50 + gradient_clip_val: 1.0 + accumulate_grad_batches: 1 + val_check_interval: 1.0 + num_workers: 4 + pin_memory: true + limit_train_batches: 1.0 + limit_val_batches: 1.0 + limit_test_batches: 1.0 + num_sanity_val_steps: 2 + log_grad_norm: true + log_param_norm: true + log_norm_every_n_steps: 50 +data: + name: synth-m + data_folder: __DATA_ROOT__/datasets/synth-m + n_var: 2 + seq_length: 128 + normalize: false + train_split: 0.8 + val_split: 0.1 +model: + name: bridge + channels: 64 + layers: 3 + nheads: 8 + dropout: 0.0 + latent_dim: 32 + num_latents: 16 + latent_unit: 1 + repre_emb_channels: 32 + prototype_dim: 32 + context_dim: 32 + text_dim: 1024 + fusion_type: gated_add + cond_drop_prob: 0.5 + use_cfg: true + model_channels: 64 + num_res_blocks: 2 + attention_resolutions: + - 1 + - 2 + - 4 + channel_mult: + - 1 + - 2 + - 4 + - 4 + num_heads: 8 + num_head_channels: -1 + use_scale_shift_norm: true + resblock_updown: true + use_spatial_transformer: true + transformer_depth: 1 + diffusion_steps: 50 + noise_schedule: quad + beta_start: 0.0001 + beta_end: 0.5 + bridge_type: vp +condition: + text: + enabled: true + input_dim: 1024 + embedding_key: cap_emb + dropout: 0.0 + cfg_scale: 1.0 + attribute: + enabled: false + continuous_dim: 0 + discrete_configs: [] + output_dim: 128 + dropout: 0.0 + label: + enabled: false + num_classes: 10 + output_dim: 64 + fusion: concat + condition_dropout: 0.0 +eval: + n_samples: 10 + metrics: + - dtw + - wape + - ed + - crps + - acd + - sd + - kd + - mdd + - fid + - sfd + - jftsd + - prdc_f1 + - joint_prdc_f1 + - cttp + - disc_auc + batch_size: 128 + save_samples: true + sampler: ddim + display_interval: 10 + clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-m/model_configs.yaml + clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-m/clip_model_best.pth + cache_folder: __CACHE_ROOT__/bridge/synth-m/seed2 + use_longalign: false + reference_split: train + disc_auc: + k_folds: 5 + epochs: 5 + batch_size: 128 + lr: 0.001 + weight_decay: 0.0001 + hidden_channels: 128 + num_layers: 3 + dropout: 0.1 + seed: 42 + num_workers: 0 + prdc: + k: 5 + max_samples: 5000 + seed: 0 + metric: euclidean + backend: auto + knn_chunk_size: 4096 + joint_enable: true + joint_weights: + ts: 1.0 + text: 1.0 + joint_normalize: standard + segment_classifier: + enable: false + segment_len: 128 + n_segments: 3 + seed: 42 + acd_max_lag: 50 + mdd_bins: 32 + use_cache: false + cache_file: predictions_cache.pkl + viz: + enable: true + k_cases: 20 + max_vars: 8 + ncols: 5 + seed: 42 + dpi: 200 + alpha: 0.25 + figsize_per_subplot: + - 3.0 + - 2.0 + output_type: pdf +output_dir: '' +created_at: '2025-12-28T18:35:18.501692' diff --git a/experiments/bridge/synth-m/seed2/results/expected_seed_metrics.json b/experiments/bridge/synth-m/seed2/results/expected_seed_metrics.json new file mode 100644 index 0000000000000000000000000000000000000000..bda237170c364e374d5e84328a842a9f31cec359 --- /dev/null +++ b/experiments/bridge/synth-m/seed2/results/expected_seed_metrics.json @@ -0,0 +1,21 @@ +{ + "dtw": 12.013450861567545, + "wape": 96.39575187473649, + "ed": 14.510944213867187, + "crps": 0.5734683085540916, + "acd": 0.04058488830924034, + "sd": 0.05335535791835336, + "kd": 0.2815472187496497, + "mdd": 0.029644704182942704, + "fid": 40.9487228029339, + "sfd": 42.799137914105, + "jftsd": 47.69404022174564, + "prdc_f1.f1": 0.2721332174920359, + "prdc_f1.precision": 0.694, + "prdc_f1.recall": 0.16925, + "joint_prdc_f1.f1": 0.35218493819128177, + "joint_prdc_f1.precision": 0.49525, + "joint_prdc_f1.recall": 0.27325, + "cttp": 19.501730834960938, + "disc_auc": 0.99973171875 +} \ No newline at end of file diff --git a/experiments/bridge/synth-m/seed2/summary.json b/experiments/bridge/synth-m/seed2/summary.json new file mode 100644 index 0000000000000000000000000000000000000000..8a716fe6ee40fe1b7b6da467fce1a436aab8bb33 --- /dev/null +++ b/experiments/bridge/synth-m/seed2/summary.json @@ -0,0 +1,5 @@ +{ + "status": "completed", + "finished_at": "2025-12-28T20:09:44.682046", + "best_checkpoint": "checkpoints/finetune/best.ckpt" +} \ No newline at end of file diff --git a/experiments/bridge/synth-u/seed0/config.template.yaml b/experiments/bridge/synth-u/seed0/config.template.yaml new file mode 100644 index 0000000000000000000000000000000000000000..64a214fcf69a241e44de20943a485d20ae366ecb --- /dev/null +++ b/experiments/bridge/synth-u/seed0/config.template.yaml @@ -0,0 +1,169 @@ +name: synth-u_bridge +description: '' +seed: 0 +device: cuda:0 +train: + stages: + - name: finetune + epochs: 700 + lr: 0.001 + use_condition: true + freeze_modules: [] + early_stopping_patience: 50 + epochs: 700 + batch_size: 256 + lr: 0.001 + weight_decay: 0.0001 + scheduler: cosine + scheduler_params: {} + early_stopping_patience: 50 + gradient_clip_val: 1.0 + accumulate_grad_batches: 1 + val_check_interval: 1.0 + num_workers: 4 + pin_memory: true + limit_train_batches: 1.0 + limit_val_batches: 1.0 + limit_test_batches: 1.0 + num_sanity_val_steps: 2 + log_grad_norm: true + log_param_norm: true + log_norm_every_n_steps: 50 +data: + name: synth-u + data_folder: __DATA_ROOT__/datasets/synth-u + n_var: 1 + seq_length: 128 + normalize: false + train_split: 0.8 + val_split: 0.1 +model: + name: bridge + channels: 64 + layers: 3 + nheads: 8 + dropout: 0.0 + latent_dim: 32 + num_latents: 16 + latent_unit: 1 + repre_emb_channels: 32 + prototype_dim: 32 + context_dim: 32 + text_dim: 1024 + fusion_type: gated_add + cond_drop_prob: 0.5 + use_cfg: true + model_channels: 64 + num_res_blocks: 2 + attention_resolutions: + - 1 + - 2 + - 4 + channel_mult: + - 1 + - 2 + - 4 + - 4 + num_heads: 8 + num_head_channels: -1 + use_scale_shift_norm: true + resblock_updown: true + use_spatial_transformer: true + transformer_depth: 1 + diffusion_steps: 50 + noise_schedule: quad + beta_start: 0.0001 + beta_end: 0.5 + bridge_type: vp +condition: + text: + enabled: true + input_dim: 1024 + embedding_key: cap_emb + dropout: 0.0 + cfg_scale: 1.0 + attribute: + enabled: false + continuous_dim: 0 + discrete_configs: [] + output_dim: 128 + dropout: 0.0 + label: + enabled: false + num_classes: 10 + output_dim: 64 + fusion: concat + condition_dropout: 0.0 +eval: + n_samples: 10 + metrics: + - dtw + - wape + - ed + - crps + - acd + - sd + - kd + - mdd + - fid + - sfd + - jftsd + - prdc_f1 + - joint_prdc_f1 + - cttp + - disc_auc + batch_size: 128 + save_samples: true + sampler: ddim + display_interval: 10 + clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-u/model_configs.yaml + clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-u/clip_model_best.pth + cache_folder: __CACHE_ROOT__/bridge/synth-u/seed0 + use_longalign: false + reference_split: train + disc_auc: + k_folds: 5 + epochs: 5 + batch_size: 128 + lr: 0.001 + weight_decay: 0.0001 + hidden_channels: 128 + num_layers: 3 + dropout: 0.1 + seed: 42 + num_workers: 0 + prdc: + k: 5 + max_samples: 5000 + seed: 0 + metric: euclidean + backend: auto + knn_chunk_size: 4096 + joint_enable: true + joint_weights: + ts: 1.0 + text: 1.0 + joint_normalize: standard + segment_classifier: + enable: false + segment_len: 128 + n_segments: 3 + seed: 42 + acd_max_lag: 50 + mdd_bins: 32 + use_cache: false + cache_file: predictions_cache.pkl + viz: + enable: true + k_cases: 20 + max_vars: 8 + ncols: 5 + seed: 42 + dpi: 200 + alpha: 0.25 + figsize_per_subplot: + - 3.0 + - 2.0 + output_type: pdf +output_dir: '' +created_at: '2025-12-28T18:38:34.831322' diff --git a/experiments/bridge/synth-u/seed0/results/expected_seed_metrics.json b/experiments/bridge/synth-u/seed0/results/expected_seed_metrics.json new file mode 100644 index 0000000000000000000000000000000000000000..9fb81496b6400ecb5020c06b4c30d4cb60180e3b --- /dev/null +++ b/experiments/bridge/synth-u/seed0/results/expected_seed_metrics.json @@ -0,0 +1,21 @@ +{ + "dtw": 6.354155136104681, + "wape": 88.65556005794502, + "ed": 9.25147834777832, + "crps": 0.5361806164317251, + "acd": 0.02120952121913433, + "sd": 0.17698199261192563, + "kd": 0.5880551470131783, + "mdd": 0.02806243896484374, + "fid": 47.42080353150922, + "sfd": 9.808309640175452, + "jftsd": 52.52638332382759, + "prdc_f1.f1": 0.14994884425918906, + "prdc_f1.precision": 0.5735, + "prdc_f1.recall": 0.08625, + "joint_prdc_f1.f1": 0.610718726016884, + "joint_prdc_f1.precision": 0.8145, + "joint_prdc_f1.recall": 0.4885, + "cttp": 23.847719635009767, + "disc_auc": 0.9995162499999999 +} \ No newline at end of file diff --git a/experiments/bridge/synth-u/seed0/summary.json b/experiments/bridge/synth-u/seed0/summary.json new file mode 100644 index 0000000000000000000000000000000000000000..d3f2c0548e000873f179d38884ed86d605984b77 --- /dev/null +++ b/experiments/bridge/synth-u/seed0/summary.json @@ -0,0 +1,5 @@ +{ + "status": "completed", + "finished_at": "2025-12-28T19:47:20.050140", + "best_checkpoint": "checkpoints/finetune/best.ckpt" +} \ No newline at end of file diff --git a/experiments/bridge/synth-u/seed1/config.template.yaml b/experiments/bridge/synth-u/seed1/config.template.yaml new file mode 100644 index 0000000000000000000000000000000000000000..5bfc87ee08a7d9dd1720b75b8795196a659a2ab5 --- /dev/null +++ b/experiments/bridge/synth-u/seed1/config.template.yaml @@ -0,0 +1,169 @@ +name: synth-u_bridge +description: '' +seed: 1 +device: cuda:0 +train: + stages: + - name: finetune + epochs: 700 + lr: 0.001 + use_condition: true + freeze_modules: [] + early_stopping_patience: 50 + epochs: 700 + batch_size: 256 + lr: 0.001 + weight_decay: 0.0001 + scheduler: cosine + scheduler_params: {} + early_stopping_patience: 50 + gradient_clip_val: 1.0 + accumulate_grad_batches: 1 + val_check_interval: 1.0 + num_workers: 4 + pin_memory: true + limit_train_batches: 1.0 + limit_val_batches: 1.0 + limit_test_batches: 1.0 + num_sanity_val_steps: 2 + log_grad_norm: true + log_param_norm: true + log_norm_every_n_steps: 50 +data: + name: synth-u + data_folder: __DATA_ROOT__/datasets/synth-u + n_var: 1 + seq_length: 128 + normalize: false + train_split: 0.8 + val_split: 0.1 +model: + name: bridge + channels: 64 + layers: 3 + nheads: 8 + dropout: 0.0 + latent_dim: 32 + num_latents: 16 + latent_unit: 1 + repre_emb_channels: 32 + prototype_dim: 32 + context_dim: 32 + text_dim: 1024 + fusion_type: gated_add + cond_drop_prob: 0.5 + use_cfg: true + model_channels: 64 + num_res_blocks: 2 + attention_resolutions: + - 1 + - 2 + - 4 + channel_mult: + - 1 + - 2 + - 4 + - 4 + num_heads: 8 + num_head_channels: -1 + use_scale_shift_norm: true + resblock_updown: true + use_spatial_transformer: true + transformer_depth: 1 + diffusion_steps: 50 + noise_schedule: quad + beta_start: 0.0001 + beta_end: 0.5 + bridge_type: vp +condition: + text: + enabled: true + input_dim: 1024 + embedding_key: cap_emb + dropout: 0.0 + cfg_scale: 1.0 + attribute: + enabled: false + continuous_dim: 0 + discrete_configs: [] + output_dim: 128 + dropout: 0.0 + label: + enabled: false + num_classes: 10 + output_dim: 64 + fusion: concat + condition_dropout: 0.0 +eval: + n_samples: 10 + metrics: + - dtw + - wape + - ed + - crps + - acd + - sd + - kd + - mdd + - fid + - sfd + - jftsd + - prdc_f1 + - joint_prdc_f1 + - cttp + - disc_auc + batch_size: 128 + save_samples: true + sampler: ddim + display_interval: 10 + clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-u/model_configs.yaml + clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-u/clip_model_best.pth + cache_folder: __CACHE_ROOT__/bridge/synth-u/seed1 + use_longalign: false + reference_split: train + disc_auc: + k_folds: 5 + epochs: 5 + batch_size: 128 + lr: 0.001 + weight_decay: 0.0001 + hidden_channels: 128 + num_layers: 3 + dropout: 0.1 + seed: 42 + num_workers: 0 + prdc: + k: 5 + max_samples: 5000 + seed: 0 + metric: euclidean + backend: auto + knn_chunk_size: 4096 + joint_enable: true + joint_weights: + ts: 1.0 + text: 1.0 + joint_normalize: standard + segment_classifier: + enable: false + segment_len: 128 + n_segments: 3 + seed: 42 + acd_max_lag: 50 + mdd_bins: 32 + use_cache: false + cache_file: predictions_cache.pkl + viz: + enable: true + k_cases: 20 + max_vars: 8 + ncols: 5 + seed: 42 + dpi: 200 + alpha: 0.25 + figsize_per_subplot: + - 3.0 + - 2.0 + output_type: pdf +output_dir: '' +created_at: '2025-12-28T18:38:34.803045' diff --git a/experiments/bridge/synth-u/seed1/results/expected_seed_metrics.json b/experiments/bridge/synth-u/seed1/results/expected_seed_metrics.json new file mode 100644 index 0000000000000000000000000000000000000000..461e38b17d6c06d20192b464e85f905a41951388 --- /dev/null +++ b/experiments/bridge/synth-u/seed1/results/expected_seed_metrics.json @@ -0,0 +1,21 @@ +{ + "dtw": 5.753590787621783, + "wape": 85.46210875970107, + "ed": 8.889996612548828, + "crps": 0.5136260835656676, + "acd": 0.028903482481837273, + "sd": 0.022280193616036857, + "kd": 0.6368266785364254, + "mdd": 0.028317220052083324, + "fid": 48.50368331516401, + "sfd": 9.037110945980103, + "jftsd": 54.73198046870306, + "prdc_f1.f1": 0.1260411555203799, + "prdc_f1.precision": 0.56075, + "prdc_f1.recall": 0.071, + "joint_prdc_f1.f1": 0.5706327191739218, + "joint_prdc_f1.precision": 0.78725, + "joint_prdc_f1.recall": 0.4475, + "cttp": 22.753153106689453, + "disc_auc": 0.9995903125 +} \ No newline at end of file diff --git a/experiments/bridge/synth-u/seed1/summary.json b/experiments/bridge/synth-u/seed1/summary.json new file mode 100644 index 0000000000000000000000000000000000000000..0ffd6a77acc2e56a8d920efbe0201150bf3a1ce1 --- /dev/null +++ b/experiments/bridge/synth-u/seed1/summary.json @@ -0,0 +1,5 @@ +{ + "status": "completed", + "finished_at": "2025-12-28T19:17:30.004892", + "best_checkpoint": "checkpoints/finetune/best.ckpt" +} \ No newline at end of file diff --git a/experiments/bridge/synth-u/seed2/config.template.yaml b/experiments/bridge/synth-u/seed2/config.template.yaml new file mode 100644 index 0000000000000000000000000000000000000000..22612b68c809668fb352b8f31399ee4aefaada1c --- /dev/null +++ b/experiments/bridge/synth-u/seed2/config.template.yaml @@ -0,0 +1,169 @@ +name: synth-u_bridge +description: '' +seed: 2 +device: cuda:0 +train: + stages: + - name: finetune + epochs: 700 + lr: 0.001 + use_condition: true + freeze_modules: [] + early_stopping_patience: 50 + epochs: 700 + batch_size: 256 + lr: 0.001 + weight_decay: 0.0001 + scheduler: cosine + scheduler_params: {} + early_stopping_patience: 50 + gradient_clip_val: 1.0 + accumulate_grad_batches: 1 + val_check_interval: 1.0 + num_workers: 4 + pin_memory: true + limit_train_batches: 1.0 + limit_val_batches: 1.0 + limit_test_batches: 1.0 + num_sanity_val_steps: 2 + log_grad_norm: true + log_param_norm: true + log_norm_every_n_steps: 50 +data: + name: synth-u + data_folder: __DATA_ROOT__/datasets/synth-u + n_var: 1 + seq_length: 128 + normalize: false + train_split: 0.8 + val_split: 0.1 +model: + name: bridge + channels: 64 + layers: 3 + nheads: 8 + dropout: 0.0 + latent_dim: 32 + num_latents: 16 + latent_unit: 1 + repre_emb_channels: 32 + prototype_dim: 32 + context_dim: 32 + text_dim: 1024 + fusion_type: gated_add + cond_drop_prob: 0.5 + use_cfg: true + model_channels: 64 + num_res_blocks: 2 + attention_resolutions: + - 1 + - 2 + - 4 + channel_mult: + - 1 + - 2 + - 4 + - 4 + num_heads: 8 + num_head_channels: -1 + use_scale_shift_norm: true + resblock_updown: true + use_spatial_transformer: true + transformer_depth: 1 + diffusion_steps: 50 + noise_schedule: quad + beta_start: 0.0001 + beta_end: 0.5 + bridge_type: vp +condition: + text: + enabled: true + input_dim: 1024 + embedding_key: cap_emb + dropout: 0.0 + cfg_scale: 1.0 + attribute: + enabled: false + continuous_dim: 0 + discrete_configs: [] + output_dim: 128 + dropout: 0.0 + label: + enabled: false + num_classes: 10 + output_dim: 64 + fusion: concat + condition_dropout: 0.0 +eval: + n_samples: 10 + metrics: + - dtw + - wape + - ed + - crps + - acd + - sd + - kd + - mdd + - fid + - sfd + - jftsd + - prdc_f1 + - joint_prdc_f1 + - cttp + - disc_auc + batch_size: 128 + save_samples: true + sampler: ddim + display_interval: 10 + clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-u/model_configs.yaml + clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-u/clip_model_best.pth + cache_folder: __CACHE_ROOT__/bridge/synth-u/seed2 + use_longalign: false + reference_split: train + disc_auc: + k_folds: 5 + epochs: 5 + batch_size: 128 + lr: 0.001 + weight_decay: 0.0001 + hidden_channels: 128 + num_layers: 3 + dropout: 0.1 + seed: 42 + num_workers: 0 + prdc: + k: 5 + max_samples: 5000 + seed: 0 + metric: euclidean + backend: auto + knn_chunk_size: 4096 + joint_enable: true + joint_weights: + ts: 1.0 + text: 1.0 + joint_normalize: standard + segment_classifier: + enable: false + segment_len: 128 + n_segments: 3 + seed: 42 + acd_max_lag: 50 + mdd_bins: 32 + use_cache: false + cache_file: predictions_cache.pkl + viz: + enable: true + k_cases: 20 + max_vars: 8 + ncols: 5 + seed: 42 + dpi: 200 + alpha: 0.25 + figsize_per_subplot: + - 3.0 + - 2.0 + output_type: pdf +output_dir: '' +created_at: '2025-12-28T18:40:44.845855' diff --git a/experiments/bridge/synth-u/seed2/results/expected_seed_metrics.json b/experiments/bridge/synth-u/seed2/results/expected_seed_metrics.json new file mode 100644 index 0000000000000000000000000000000000000000..9837fc96c331efd30a2bff155556aa034384234f --- /dev/null +++ b/experiments/bridge/synth-u/seed2/results/expected_seed_metrics.json @@ -0,0 +1,21 @@ +{ + "dtw": 5.530313082307284, + "wape": 82.54036397922455, + "ed": 8.522937805175781, + "crps": 0.5017315307045973, + "acd": 0.04407091811299324, + "sd": 0.0928177492640565, + "kd": 0.7001046294738782, + "mdd": 0.028313680013020828, + "fid": 54.61961133293579, + "sfd": 9.44118332955128, + "jftsd": 58.16182762137342, + "prdc_f1.f1": 0.1200266429840142, + "prdc_f1.precision": 0.6375, + "prdc_f1.recall": 0.06625, + "joint_prdc_f1.f1": 0.6256576958525346, + "joint_prdc_f1.precision": 0.86675, + "joint_prdc_f1.recall": 0.4895, + "cttp": 22.37494738769531, + "disc_auc": 0.99950046875 +} \ No newline at end of file diff --git a/experiments/bridge/synth-u/seed2/summary.json b/experiments/bridge/synth-u/seed2/summary.json new file mode 100644 index 0000000000000000000000000000000000000000..ff8c17ba7f793e1bb656e682a5f3e665eba9edae --- /dev/null +++ b/experiments/bridge/synth-u/seed2/summary.json @@ -0,0 +1,5 @@ +{ + "status": "completed", + "finished_at": "2025-12-28T20:17:12.870391", + "best_checkpoint": "checkpoints/finetune/best.ckpt" +} \ No newline at end of file diff --git a/experiments/diffusets/synth-m/seed0/config.template.yaml b/experiments/diffusets/synth-m/seed0/config.template.yaml new file mode 100644 index 0000000000000000000000000000000000000000..cd094954469fa0680e524a102d5aae98b2a9a6aa --- /dev/null +++ b/experiments/diffusets/synth-m/seed0/config.template.yaml @@ -0,0 +1,156 @@ +name: synth-m_diffusets +description: '' +seed: 0 +device: cuda:0 +train: + stages: + - name: vae_pretrain + epochs: 200 + lr: 0.001 + use_condition: false + freeze_modules: [] + early_stopping_patience: 50 + - name: finetune + epochs: 500 + lr: 0.001 + use_condition: true + freeze_modules: [] + load_from_stage: vae_pretrain + early_stopping_patience: 50 + epochs: 700 + batch_size: 256 + lr: 0.001 + weight_decay: 0.0001 + scheduler: cosine + scheduler_params: {} + early_stopping_patience: 50 + gradient_clip_val: 1.0 + accumulate_grad_batches: 1 + val_check_interval: 1.0 + num_workers: 4 + pin_memory: true + limit_train_batches: 1.0 + limit_val_batches: 1.0 + limit_test_batches: 1.0 + num_sanity_val_steps: 2 + log_grad_norm: true + log_param_norm: true + log_norm_every_n_steps: 50 +data: + name: synth-m + data_folder: __DATA_ROOT__/datasets/synth-m + n_var: 2 + seq_length: 128 + normalize: false + train_split: 0.8 + val_split: 0.1 +model: + name: diffusets + channels: 64 + layers: 3 + nheads: 8 + dropout: 0.1 + latent_channels: 4 + diffusion_steps: 50 + noise_schedule: quad + beta_start: 0.0001 + beta_end: 0.5 + kernel_size: 3 + num_levels: 5 + kld_weight: 1.0 + kld_annealing: true + kld_annealing_type: linear + kld_warmup_epochs: 300 + kld_start_weight: 0.01 +condition: + text: + enabled: true + input_dim: 1024 + embedding_key: cap_emb + dropout: 0.0 + cfg_scale: 1.0 + attribute: + enabled: false + continuous_dim: 0 + discrete_configs: [] + output_dim: 128 + dropout: 0.0 + label: + enabled: false + num_classes: 10 + output_dim: 64 + fusion: concat + condition_dropout: 0.0 +eval: + n_samples: 10 + metrics: + - dtw + - wape + - ed + - crps + - acd + - sd + - kd + - mdd + - fid + - sfd + - jftsd + - prdc_f1 + - joint_prdc_f1 + - cttp + - disc_auc + batch_size: 128 + save_samples: true + sampler: ddim + display_interval: 10 + clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-m/model_configs.yaml + clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-m/clip_model_best.pth + cache_folder: __CACHE_ROOT__/diffusets/synth-m/seed0 + use_longalign: false + reference_split: train + disc_auc: + k_folds: 5 + epochs: 5 + batch_size: 128 + lr: 0.001 + weight_decay: 0.0001 + hidden_channels: 128 + num_layers: 3 + dropout: 0.1 + seed: 42 + num_workers: 0 + prdc: + k: 5 + max_samples: 5000 + seed: 0 + metric: euclidean + backend: auto + knn_chunk_size: 4096 + joint_enable: true + joint_weights: + ts: 1.0 + text: 1.0 + joint_normalize: standard + segment_classifier: + enable: false + segment_len: 128 + n_segments: 3 + seed: 42 + acd_max_lag: 50 + mdd_bins: 32 + use_cache: false + cache_file: predictions_cache.pkl + viz: + enable: true + k_cases: 20 + max_vars: 8 + ncols: 5 + seed: 42 + dpi: 200 + alpha: 0.25 + figsize_per_subplot: + - 3.0 + - 2.0 + output_type: pdf +output_dir: '' +created_at: '2025-12-28T15:52:00.016312' diff --git a/experiments/diffusets/synth-m/seed0/results/expected_seed_metrics.json b/experiments/diffusets/synth-m/seed0/results/expected_seed_metrics.json new file mode 100644 index 0000000000000000000000000000000000000000..d10e327a16ddcf580a400704c621096599544111 --- /dev/null +++ b/experiments/diffusets/synth-m/seed0/results/expected_seed_metrics.json @@ -0,0 +1,21 @@ +{ + "dtw": 11.140782107069837, + "wape": 89.60107209439903, + "ed": 13.401970443725586, + "crps": 0.48672253132439686, + "acd": 0.03553586080670357, + "sd": 0.16719546573763044, + "kd": 0.07568088321032929, + "mdd": 0.023467702229817703, + "fid": 42.76190316351483, + "sfd": 85.36533673707893, + "jftsd": 52.20228797971308, + "prdc_f1.f1": 0.2579129526462396, + "prdc_f1.precision": 0.5495, + "prdc_f1.recall": 0.1685, + "joint_prdc_f1.f1": 0.305841307814992, + "joint_prdc_f1.precision": 0.3625, + "joint_prdc_f1.recall": 0.2645, + "cttp": 16.67788977050781, + "disc_auc": 0.9999359375000001 +} \ No newline at end of file diff --git a/experiments/diffusets/synth-m/seed0/summary.json b/experiments/diffusets/synth-m/seed0/summary.json new file mode 100644 index 0000000000000000000000000000000000000000..cc32257ac6427173f9b197d4c3655dc0c98fa62f --- /dev/null +++ b/experiments/diffusets/synth-m/seed0/summary.json @@ -0,0 +1,5 @@ +{ + "status": "completed", + "finished_at": "2025-12-28T16:24:00.801438", + "best_checkpoint": "checkpoints/finetune/best.ckpt" +} \ No newline at end of file diff --git a/experiments/diffusets/synth-m/seed1/config.template.yaml b/experiments/diffusets/synth-m/seed1/config.template.yaml new file mode 100644 index 0000000000000000000000000000000000000000..83c6cccd28172b2613744d3efc74ae9396e4053e --- /dev/null +++ b/experiments/diffusets/synth-m/seed1/config.template.yaml @@ -0,0 +1,156 @@ +name: synth-m_diffusets +description: '' +seed: 1 +device: cuda:0 +train: + stages: + - name: vae_pretrain + epochs: 200 + lr: 0.001 + use_condition: false + freeze_modules: [] + early_stopping_patience: 50 + - name: finetune + epochs: 500 + lr: 0.001 + use_condition: true + freeze_modules: [] + load_from_stage: vae_pretrain + early_stopping_patience: 50 + epochs: 700 + batch_size: 256 + lr: 0.001 + weight_decay: 0.0001 + scheduler: cosine + scheduler_params: {} + early_stopping_patience: 50 + gradient_clip_val: 1.0 + accumulate_grad_batches: 1 + val_check_interval: 1.0 + num_workers: 4 + pin_memory: true + limit_train_batches: 1.0 + limit_val_batches: 1.0 + limit_test_batches: 1.0 + num_sanity_val_steps: 2 + log_grad_norm: true + log_param_norm: true + log_norm_every_n_steps: 50 +data: + name: synth-m + data_folder: __DATA_ROOT__/datasets/synth-m + n_var: 2 + seq_length: 128 + normalize: false + train_split: 0.8 + val_split: 0.1 +model: + name: diffusets + channels: 64 + layers: 3 + nheads: 8 + dropout: 0.1 + latent_channels: 4 + diffusion_steps: 50 + noise_schedule: quad + beta_start: 0.0001 + beta_end: 0.5 + kernel_size: 3 + num_levels: 5 + kld_weight: 1.0 + kld_annealing: true + kld_annealing_type: linear + kld_warmup_epochs: 300 + kld_start_weight: 0.01 +condition: + text: + enabled: true + input_dim: 1024 + embedding_key: cap_emb + dropout: 0.0 + cfg_scale: 1.0 + attribute: + enabled: false + continuous_dim: 0 + discrete_configs: [] + output_dim: 128 + dropout: 0.0 + label: + enabled: false + num_classes: 10 + output_dim: 64 + fusion: concat + condition_dropout: 0.0 +eval: + n_samples: 10 + metrics: + - dtw + - wape + - ed + - crps + - acd + - sd + - kd + - mdd + - fid + - sfd + - jftsd + - prdc_f1 + - joint_prdc_f1 + - cttp + - disc_auc + batch_size: 128 + save_samples: true + sampler: ddim + display_interval: 10 + clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-m/model_configs.yaml + clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-m/clip_model_best.pth + cache_folder: __CACHE_ROOT__/diffusets/synth-m/seed1 + use_longalign: false + reference_split: train + disc_auc: + k_folds: 5 + epochs: 5 + batch_size: 128 + lr: 0.001 + weight_decay: 0.0001 + hidden_channels: 128 + num_layers: 3 + dropout: 0.1 + seed: 42 + num_workers: 0 + prdc: + k: 5 + max_samples: 5000 + seed: 0 + metric: euclidean + backend: auto + knn_chunk_size: 4096 + joint_enable: true + joint_weights: + ts: 1.0 + text: 1.0 + joint_normalize: standard + segment_classifier: + enable: false + segment_len: 128 + n_segments: 3 + seed: 42 + acd_max_lag: 50 + mdd_bins: 32 + use_cache: false + cache_file: predictions_cache.pkl + viz: + enable: true + k_cases: 20 + max_vars: 8 + ncols: 5 + seed: 42 + dpi: 200 + alpha: 0.25 + figsize_per_subplot: + - 3.0 + - 2.0 + output_type: pdf +output_dir: '' +created_at: '2025-12-28T15:55:48.457922' diff --git a/experiments/diffusets/synth-m/seed1/results/expected_seed_metrics.json b/experiments/diffusets/synth-m/seed1/results/expected_seed_metrics.json new file mode 100644 index 0000000000000000000000000000000000000000..289f4646038f34866a9077b0f3e5d4d997df6ee1 --- /dev/null +++ b/experiments/diffusets/synth-m/seed1/results/expected_seed_metrics.json @@ -0,0 +1,21 @@ +{ + "dtw": 10.334665453160229, + "wape": 85.56766340774767, + "ed": 12.78905828857422, + "crps": 0.4503502460532916, + "acd": 0.025928545743227005, + "sd": 0.0435979218540752, + "kd": 0.08138287083778217, + "mdd": 0.021443400065104163, + "fid": 33.25484026734239, + "sfd": 79.12128260221301, + "jftsd": 39.5227926010414, + "prdc_f1.f1": 0.4241700302724521, + "prdc_f1.precision": 0.6835, + "prdc_f1.recall": 0.3075, + "joint_prdc_f1.f1": 0.5467748411978223, + "joint_prdc_f1.precision": 0.59925, + "joint_prdc_f1.recall": 0.50275, + "cttp": 21.062711242675782, + "disc_auc": 0.9997506249999999 +} \ No newline at end of file diff --git a/experiments/diffusets/synth-m/seed1/summary.json b/experiments/diffusets/synth-m/seed1/summary.json new file mode 100644 index 0000000000000000000000000000000000000000..d81e9516874816b912246d1a21af42bd619c2142 --- /dev/null +++ b/experiments/diffusets/synth-m/seed1/summary.json @@ -0,0 +1,5 @@ +{ + "status": "completed", + "finished_at": "2025-12-28T16:57:34.102696", + "best_checkpoint": "checkpoints/finetune/best.ckpt" +} \ No newline at end of file diff --git a/experiments/diffusets/synth-m/seed2/config.template.yaml b/experiments/diffusets/synth-m/seed2/config.template.yaml new file mode 100644 index 0000000000000000000000000000000000000000..64c17ecff01d1eadb3238947567876bf9cf92354 --- /dev/null +++ b/experiments/diffusets/synth-m/seed2/config.template.yaml @@ -0,0 +1,156 @@ +name: synth-m_diffusets +description: '' +seed: 2 +device: cuda:0 +train: + stages: + - name: vae_pretrain + epochs: 200 + lr: 0.001 + use_condition: false + freeze_modules: [] + early_stopping_patience: 50 + - name: finetune + epochs: 500 + lr: 0.001 + use_condition: true + freeze_modules: [] + load_from_stage: vae_pretrain + early_stopping_patience: 50 + epochs: 700 + batch_size: 256 + lr: 0.001 + weight_decay: 0.0001 + scheduler: cosine + scheduler_params: {} + early_stopping_patience: 50 + gradient_clip_val: 1.0 + accumulate_grad_batches: 1 + val_check_interval: 1.0 + num_workers: 4 + pin_memory: true + limit_train_batches: 1.0 + limit_val_batches: 1.0 + limit_test_batches: 1.0 + num_sanity_val_steps: 2 + log_grad_norm: true + log_param_norm: true + log_norm_every_n_steps: 50 +data: + name: synth-m + data_folder: __DATA_ROOT__/datasets/synth-m + n_var: 2 + seq_length: 128 + normalize: false + train_split: 0.8 + val_split: 0.1 +model: + name: diffusets + channels: 64 + layers: 3 + nheads: 8 + dropout: 0.1 + latent_channels: 4 + diffusion_steps: 50 + noise_schedule: quad + beta_start: 0.0001 + beta_end: 0.5 + kernel_size: 3 + num_levels: 5 + kld_weight: 1.0 + kld_annealing: true + kld_annealing_type: linear + kld_warmup_epochs: 300 + kld_start_weight: 0.01 +condition: + text: + enabled: true + input_dim: 1024 + embedding_key: cap_emb + dropout: 0.0 + cfg_scale: 1.0 + attribute: + enabled: false + continuous_dim: 0 + discrete_configs: [] + output_dim: 128 + dropout: 0.0 + label: + enabled: false + num_classes: 10 + output_dim: 64 + fusion: concat + condition_dropout: 0.0 +eval: + n_samples: 10 + metrics: + - dtw + - wape + - ed + - crps + - acd + - sd + - kd + - mdd + - fid + - sfd + - jftsd + - prdc_f1 + - joint_prdc_f1 + - cttp + - disc_auc + batch_size: 128 + save_samples: true + sampler: ddim + display_interval: 10 + clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-m/model_configs.yaml + clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-m/clip_model_best.pth + cache_folder: __CACHE_ROOT__/diffusets/synth-m/seed2 + use_longalign: false + reference_split: train + disc_auc: + k_folds: 5 + epochs: 5 + batch_size: 128 + lr: 0.001 + weight_decay: 0.0001 + hidden_channels: 128 + num_layers: 3 + dropout: 0.1 + seed: 42 + num_workers: 0 + prdc: + k: 5 + max_samples: 5000 + seed: 0 + metric: euclidean + backend: auto + knn_chunk_size: 4096 + joint_enable: true + joint_weights: + ts: 1.0 + text: 1.0 + joint_normalize: standard + segment_classifier: + enable: false + segment_len: 128 + n_segments: 3 + seed: 42 + acd_max_lag: 50 + mdd_bins: 32 + use_cache: false + cache_file: predictions_cache.pkl + viz: + enable: true + k_cases: 20 + max_vars: 8 + ncols: 5 + seed: 42 + dpi: 200 + alpha: 0.25 + figsize_per_subplot: + - 3.0 + - 2.0 + output_type: pdf +output_dir: '' +created_at: '2025-12-28T16:18:09.179981' diff --git a/experiments/diffusets/synth-m/seed2/results/expected_seed_metrics.json b/experiments/diffusets/synth-m/seed2/results/expected_seed_metrics.json new file mode 100644 index 0000000000000000000000000000000000000000..d901e190cb641712e8d09e45fa22a4457ddf5559 --- /dev/null +++ b/experiments/diffusets/synth-m/seed2/results/expected_seed_metrics.json @@ -0,0 +1,21 @@ +{ + "dtw": 9.317690117480163, + "wape": 81.66055296593296, + "ed": 12.043457717895508, + "crps": 0.44030666400529395, + "acd": 0.06254220008850098, + "sd": 0.04333788460742726, + "kd": 0.3218432933887936, + "mdd": 0.01764912923177083, + "fid": 30.94725361796202, + "sfd": 44.83966561329548, + "jftsd": 34.4846521694688, + "prdc_f1.f1": 0.4201174869853289, + "prdc_f1.precision": 0.76725, + "prdc_f1.recall": 0.28925, + "joint_prdc_f1.f1": 0.7147140649149922, + "joint_prdc_f1.precision": 0.82575, + "joint_prdc_f1.recall": 0.63, + "cttp": 25.972833389282226, + "disc_auc": 0.9997812500000001 +} \ No newline at end of file diff --git a/experiments/diffusets/synth-m/seed2/summary.json b/experiments/diffusets/synth-m/seed2/summary.json new file mode 100644 index 0000000000000000000000000000000000000000..dbb962a73877a08a51837d54a83a79db3ec6ff82 --- /dev/null +++ b/experiments/diffusets/synth-m/seed2/summary.json @@ -0,0 +1,5 @@ +{ + "status": "completed", + "finished_at": "2025-12-28T17:31:42.961739", + "best_checkpoint": "checkpoints/finetune/best.ckpt" +} \ No newline at end of file diff --git a/experiments/diffusets/synth-u/seed0/config.template.yaml b/experiments/diffusets/synth-u/seed0/config.template.yaml new file mode 100644 index 0000000000000000000000000000000000000000..ca235ad7fc87802e8482ff1e6e56614ff141039b --- /dev/null +++ b/experiments/diffusets/synth-u/seed0/config.template.yaml @@ -0,0 +1,156 @@ +name: synth-u_diffusets +description: '' +seed: 0 +device: cuda:0 +train: + stages: + - name: vae_pretrain + epochs: 200 + lr: 0.001 + use_condition: false + freeze_modules: [] + early_stopping_patience: 50 + - name: finetune + epochs: 500 + lr: 0.001 + use_condition: true + freeze_modules: [] + load_from_stage: vae_pretrain + early_stopping_patience: 50 + epochs: 700 + batch_size: 256 + lr: 0.001 + weight_decay: 0.0001 + scheduler: cosine + scheduler_params: {} + early_stopping_patience: 50 + gradient_clip_val: 1.0 + accumulate_grad_batches: 1 + val_check_interval: 1.0 + num_workers: 4 + pin_memory: true + limit_train_batches: 1.0 + limit_val_batches: 1.0 + limit_test_batches: 1.0 + num_sanity_val_steps: 2 + log_grad_norm: true + log_param_norm: true + log_norm_every_n_steps: 50 +data: + name: synth-u + data_folder: __DATA_ROOT__/datasets/synth-u + n_var: 1 + seq_length: 128 + normalize: false + train_split: 0.8 + val_split: 0.1 +model: + name: diffusets + channels: 64 + layers: 3 + nheads: 8 + dropout: 0.1 + latent_channels: 4 + diffusion_steps: 50 + noise_schedule: quad + beta_start: 0.0001 + beta_end: 0.5 + kernel_size: 3 + num_levels: 5 + kld_weight: 1.0 + kld_annealing: true + kld_annealing_type: linear + kld_warmup_epochs: 300 + kld_start_weight: 0.01 +condition: + text: + enabled: true + input_dim: 1024 + embedding_key: cap_emb + dropout: 0.0 + cfg_scale: 1.0 + attribute: + enabled: false + continuous_dim: 0 + discrete_configs: [] + output_dim: 128 + dropout: 0.0 + label: + enabled: false + num_classes: 10 + output_dim: 64 + fusion: concat + condition_dropout: 0.0 +eval: + n_samples: 10 + metrics: + - dtw + - wape + - ed + - crps + - acd + - sd + - kd + - mdd + - fid + - sfd + - jftsd + - prdc_f1 + - joint_prdc_f1 + - cttp + - disc_auc + batch_size: 128 + save_samples: true + sampler: ddim + display_interval: 10 + clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-u/model_configs.yaml + clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-u/clip_model_best.pth + cache_folder: __CACHE_ROOT__/diffusets/synth-u/seed0 + use_longalign: false + reference_split: train + disc_auc: + k_folds: 5 + epochs: 5 + batch_size: 128 + lr: 0.001 + weight_decay: 0.0001 + hidden_channels: 128 + num_layers: 3 + dropout: 0.1 + seed: 42 + num_workers: 0 + prdc: + k: 5 + max_samples: 5000 + seed: 0 + metric: euclidean + backend: auto + knn_chunk_size: 4096 + joint_enable: true + joint_weights: + ts: 1.0 + text: 1.0 + joint_normalize: standard + segment_classifier: + enable: false + segment_len: 128 + n_segments: 3 + seed: 42 + acd_max_lag: 50 + mdd_bins: 32 + use_cache: false + cache_file: predictions_cache.pkl + viz: + enable: true + k_cases: 20 + max_vars: 8 + ncols: 5 + seed: 42 + dpi: 200 + alpha: 0.25 + figsize_per_subplot: + - 3.0 + - 2.0 + output_type: pdf +output_dir: '' +created_at: '2025-12-28T16:18:09.184954' diff --git a/experiments/diffusets/synth-u/seed0/results/expected_seed_metrics.json b/experiments/diffusets/synth-u/seed0/results/expected_seed_metrics.json new file mode 100644 index 0000000000000000000000000000000000000000..4fc4bb38260bcc25f4802381a36896872dd02473 --- /dev/null +++ b/experiments/diffusets/synth-u/seed0/results/expected_seed_metrics.json @@ -0,0 +1,21 @@ +{ + "dtw": 6.532428591720535, + "wape": 87.80967145497634, + "ed": 9.246404472351074, + "crps": 0.4526067496610641, + "acd": 0.0433688648045063, + "sd": 0.2959921614171813, + "kd": 1.8287045933291228, + "mdd": 0.0260916748046875, + "fid": 51.36839796221051, + "sfd": 54.6720668718775, + "jftsd": 58.86045523270212, + "prdc_f1.f1": 0.3003663853727145, + "prdc_f1.precision": 0.4955, + "prdc_f1.recall": 0.2155, + "joint_prdc_f1.f1": 0.5770832266325224, + "joint_prdc_f1.precision": 0.657, + "joint_prdc_f1.recall": 0.5145, + "cttp": 19.388934158325196, + "disc_auc": 0.999903125 +} \ No newline at end of file diff --git a/experiments/diffusets/synth-u/seed0/summary.json b/experiments/diffusets/synth-u/seed0/summary.json new file mode 100644 index 0000000000000000000000000000000000000000..240bf255a8d30b6f09fd6b0e275d51ce8c8ab0e0 --- /dev/null +++ b/experiments/diffusets/synth-u/seed0/summary.json @@ -0,0 +1,5 @@ +{ + "status": "completed", + "finished_at": "2025-12-28T17:39:02.597689", + "best_checkpoint": "checkpoints/finetune/best.ckpt" +} \ No newline at end of file diff --git a/experiments/diffusets/synth-u/seed1/config.template.yaml b/experiments/diffusets/synth-u/seed1/config.template.yaml new file mode 100644 index 0000000000000000000000000000000000000000..0af4f54568981db8255e45e6a5d47d44915bce9e --- /dev/null +++ b/experiments/diffusets/synth-u/seed1/config.template.yaml @@ -0,0 +1,156 @@ +name: synth-u_diffusets +description: '' +seed: 1 +device: cuda:0 +train: + stages: + - name: vae_pretrain + epochs: 200 + lr: 0.001 + use_condition: false + freeze_modules: [] + early_stopping_patience: 50 + - name: finetune + epochs: 500 + lr: 0.001 + use_condition: true + freeze_modules: [] + load_from_stage: vae_pretrain + early_stopping_patience: 50 + epochs: 700 + batch_size: 256 + lr: 0.001 + weight_decay: 0.0001 + scheduler: cosine + scheduler_params: {} + early_stopping_patience: 50 + gradient_clip_val: 1.0 + accumulate_grad_batches: 1 + val_check_interval: 1.0 + num_workers: 4 + pin_memory: true + limit_train_batches: 1.0 + limit_val_batches: 1.0 + limit_test_batches: 1.0 + num_sanity_val_steps: 2 + log_grad_norm: true + log_param_norm: true + log_norm_every_n_steps: 50 +data: + name: synth-u + data_folder: __DATA_ROOT__/datasets/synth-u + n_var: 1 + seq_length: 128 + normalize: false + train_split: 0.8 + val_split: 0.1 +model: + name: diffusets + channels: 64 + layers: 3 + nheads: 8 + dropout: 0.1 + latent_channels: 4 + diffusion_steps: 50 + noise_schedule: quad + beta_start: 0.0001 + beta_end: 0.5 + kernel_size: 3 + num_levels: 5 + kld_weight: 1.0 + kld_annealing: true + kld_annealing_type: linear + kld_warmup_epochs: 300 + kld_start_weight: 0.01 +condition: + text: + enabled: true + input_dim: 1024 + embedding_key: cap_emb + dropout: 0.0 + cfg_scale: 1.0 + attribute: + enabled: false + continuous_dim: 0 + discrete_configs: [] + output_dim: 128 + dropout: 0.0 + label: + enabled: false + num_classes: 10 + output_dim: 64 + fusion: concat + condition_dropout: 0.0 +eval: + n_samples: 10 + metrics: + - dtw + - wape + - ed + - crps + - acd + - sd + - kd + - mdd + - fid + - sfd + - jftsd + - prdc_f1 + - joint_prdc_f1 + - cttp + - disc_auc + batch_size: 128 + save_samples: true + sampler: ddim + display_interval: 10 + clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-u/model_configs.yaml + clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-u/clip_model_best.pth + cache_folder: __CACHE_ROOT__/diffusets/synth-u/seed1 + use_longalign: false + reference_split: train + disc_auc: + k_folds: 5 + epochs: 5 + batch_size: 128 + lr: 0.001 + weight_decay: 0.0001 + hidden_channels: 128 + num_layers: 3 + dropout: 0.1 + seed: 42 + num_workers: 0 + prdc: + k: 5 + max_samples: 5000 + seed: 0 + metric: euclidean + backend: auto + knn_chunk_size: 4096 + joint_enable: true + joint_weights: + ts: 1.0 + text: 1.0 + joint_normalize: standard + segment_classifier: + enable: false + segment_len: 128 + n_segments: 3 + seed: 42 + acd_max_lag: 50 + mdd_bins: 32 + use_cache: false + cache_file: predictions_cache.pkl + viz: + enable: true + k_cases: 20 + max_vars: 8 + ncols: 5 + seed: 42 + dpi: 200 + alpha: 0.25 + figsize_per_subplot: + - 3.0 + - 2.0 + output_type: pdf +output_dir: '' +created_at: '2025-12-28T16:18:09.185009' diff --git a/experiments/diffusets/synth-u/seed1/results/expected_seed_metrics.json b/experiments/diffusets/synth-u/seed1/results/expected_seed_metrics.json new file mode 100644 index 0000000000000000000000000000000000000000..6b7ac6c4016db64721c9a155febd0a66567e879c --- /dev/null +++ b/experiments/diffusets/synth-u/seed1/results/expected_seed_metrics.json @@ -0,0 +1,21 @@ +{ + "dtw": 4.954761916935516, + "wape": 81.29904120650428, + "ed": 8.348238151550293, + "crps": 0.4236568123287523, + "acd": 0.05647331476211548, + "sd": 0.031882725015297264, + "kd": 0.2726460478386823, + "mdd": 0.01779575602213541, + "fid": 33.97250146771219, + "sfd": 7.695950143207412, + "jftsd": 37.68384150873891, + "prdc_f1.f1": 0.33002626143405134, + "prdc_f1.precision": 0.62275, + "prdc_f1.recall": 0.2245, + "joint_prdc_f1.f1": 0.8377482661944814, + "joint_prdc_f1.precision": 0.93625, + "joint_prdc_f1.recall": 0.758, + "cttp": 27.93660757446289, + "disc_auc": 0.9999784375 +} \ No newline at end of file diff --git a/experiments/diffusets/synth-u/seed1/summary.json b/experiments/diffusets/synth-u/seed1/summary.json new file mode 100644 index 0000000000000000000000000000000000000000..0f38dd17eefff677a350e22522d6cf2898b8a57c --- /dev/null +++ b/experiments/diffusets/synth-u/seed1/summary.json @@ -0,0 +1,5 @@ +{ + "status": "completed", + "finished_at": "2025-12-28T17:34:41.687120", + "best_checkpoint": "checkpoints/finetune/best.ckpt" +} \ No newline at end of file diff --git a/experiments/diffusets/synth-u/seed2/config.template.yaml b/experiments/diffusets/synth-u/seed2/config.template.yaml new file mode 100644 index 0000000000000000000000000000000000000000..091cb14335c040198a33648216790eee8bd47046 --- /dev/null +++ b/experiments/diffusets/synth-u/seed2/config.template.yaml @@ -0,0 +1,156 @@ +name: synth-u_diffusets +description: '' +seed: 2 +device: cuda:0 +train: + stages: + - name: vae_pretrain + epochs: 200 + lr: 0.001 + use_condition: false + freeze_modules: [] + early_stopping_patience: 50 + - name: finetune + epochs: 500 + lr: 0.001 + use_condition: true + freeze_modules: [] + load_from_stage: vae_pretrain + early_stopping_patience: 50 + epochs: 700 + batch_size: 256 + lr: 0.001 + weight_decay: 0.0001 + scheduler: cosine + scheduler_params: {} + early_stopping_patience: 50 + gradient_clip_val: 1.0 + accumulate_grad_batches: 1 + val_check_interval: 1.0 + num_workers: 4 + pin_memory: true + limit_train_batches: 1.0 + limit_val_batches: 1.0 + limit_test_batches: 1.0 + num_sanity_val_steps: 2 + log_grad_norm: true + log_param_norm: true + log_norm_every_n_steps: 50 +data: + name: synth-u + data_folder: __DATA_ROOT__/datasets/synth-u + n_var: 1 + seq_length: 128 + normalize: false + train_split: 0.8 + val_split: 0.1 +model: + name: diffusets + channels: 64 + layers: 3 + nheads: 8 + dropout: 0.1 + latent_channels: 4 + diffusion_steps: 50 + noise_schedule: quad + beta_start: 0.0001 + beta_end: 0.5 + kernel_size: 3 + num_levels: 5 + kld_weight: 1.0 + kld_annealing: true + kld_annealing_type: linear + kld_warmup_epochs: 300 + kld_start_weight: 0.01 +condition: + text: + enabled: true + input_dim: 1024 + embedding_key: cap_emb + dropout: 0.0 + cfg_scale: 1.0 + attribute: + enabled: false + continuous_dim: 0 + discrete_configs: [] + output_dim: 128 + dropout: 0.0 + label: + enabled: false + num_classes: 10 + output_dim: 64 + fusion: concat + condition_dropout: 0.0 +eval: + n_samples: 10 + metrics: + - dtw + - wape + - ed + - crps + - acd + - sd + - kd + - mdd + - fid + - sfd + - jftsd + - prdc_f1 + - joint_prdc_f1 + - cttp + - disc_auc + batch_size: 128 + save_samples: true + sampler: ddim + display_interval: 10 + clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-u/model_configs.yaml + clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-u/clip_model_best.pth + cache_folder: __CACHE_ROOT__/diffusets/synth-u/seed2 + use_longalign: false + reference_split: train + disc_auc: + k_folds: 5 + epochs: 5 + batch_size: 128 + lr: 0.001 + weight_decay: 0.0001 + hidden_channels: 128 + num_layers: 3 + dropout: 0.1 + seed: 42 + num_workers: 0 + prdc: + k: 5 + max_samples: 5000 + seed: 0 + metric: euclidean + backend: auto + knn_chunk_size: 4096 + joint_enable: true + joint_weights: + ts: 1.0 + text: 1.0 + joint_normalize: standard + segment_classifier: + enable: false + segment_len: 128 + n_segments: 3 + seed: 42 + acd_max_lag: 50 + mdd_bins: 32 + use_cache: false + cache_file: predictions_cache.pkl + viz: + enable: true + k_cases: 20 + max_vars: 8 + ncols: 5 + seed: 42 + dpi: 200 + alpha: 0.25 + figsize_per_subplot: + - 3.0 + - 2.0 + output_type: pdf +output_dir: '' +created_at: '2025-12-28T16:59:44.641212' diff --git a/experiments/diffusets/synth-u/seed2/results/expected_seed_metrics.json b/experiments/diffusets/synth-u/seed2/results/expected_seed_metrics.json new file mode 100644 index 0000000000000000000000000000000000000000..f6f558786a13f3c7a916a45cd29ae05f1a05141b --- /dev/null +++ b/experiments/diffusets/synth-u/seed2/results/expected_seed_metrics.json @@ -0,0 +1,21 @@ +{ + "dtw": 5.604377370645617, + "wape": 89.19720450918459, + "ed": 9.010427093505859, + "crps": 0.4444298378533488, + "acd": 0.06584841758012772, + "sd": 0.0934342422658183, + "kd": 0.3979876287319106, + "mdd": 0.017146341959635412, + "fid": 39.433125114674, + "sfd": 9.23097797963554, + "jftsd": 43.144632222561086, + "prdc_f1.f1": 0.275845928439235, + "prdc_f1.precision": 0.63425, + "prdc_f1.recall": 0.17625, + "joint_prdc_f1.f1": 0.805401887654133, + "joint_prdc_f1.precision": 0.93475, + "joint_prdc_f1.recall": 0.7075, + "cttp": 26.66297784423828, + "disc_auc": 0.9999381249999999 +} \ No newline at end of file diff --git a/experiments/diffusets/synth-u/seed2/summary.json b/experiments/diffusets/synth-u/seed2/summary.json new file mode 100644 index 0000000000000000000000000000000000000000..86d8cfbfc3efa72071fce10d62210373ac07c276 --- /dev/null +++ b/experiments/diffusets/synth-u/seed2/summary.json @@ -0,0 +1,5 @@ +{ + "status": "completed", + "finished_at": "2025-12-28T17:53:54.368325", + "best_checkpoint": "checkpoints/finetune/best.ckpt" +} \ No newline at end of file diff --git a/experiments/t2s/synth-m/seed0/config.template.yaml b/experiments/t2s/synth-m/seed0/config.template.yaml new file mode 100644 index 0000000000000000000000000000000000000000..0026511a6c5b447346aefa9a594a0451f0c1e96c --- /dev/null +++ b/experiments/t2s/synth-m/seed0/config.template.yaml @@ -0,0 +1,153 @@ +name: synth-m_t2s +description: '' +seed: 0 +device: cuda:0 +train: + stages: + - name: ae_pretrain + epochs: 200 + lr: 0.001 + use_condition: false + freeze_modules: [] + early_stopping_patience: 50 + - name: finetune + epochs: 500 + lr: 0.001 + use_condition: true + freeze_modules: [] + load_from_stage: ae_pretrain + early_stopping_patience: 50 + epochs: 700 + batch_size: 256 + lr: 0.001 + weight_decay: 0.0001 + scheduler: cosine + scheduler_params: {} + early_stopping_patience: 50 + gradient_clip_val: 1.0 + accumulate_grad_batches: 1 + val_check_interval: 1.0 + num_workers: 4 + pin_memory: true + limit_train_batches: 1.0 + limit_val_batches: 1.0 + limit_test_batches: 1.0 + num_sanity_val_steps: 2 + log_grad_norm: true + log_param_norm: true + log_norm_every_n_steps: 50 +data: + name: synth-m + data_folder: __DATA_ROOT__/datasets/synth-m + n_var: 2 + seq_length: 128 + normalize: false + train_split: 0.8 + val_split: 0.1 +model: + name: t2s + channels: 64 + layers: 4 + nheads: 4 + dropout: 0.1 + patch_size: 2 + mlp_ratio: 2.0 + flow_steps: 10 + cfg_scale: 9.0 + ae: + num_hiddens: 128 + num_residual_layers: 2 + num_residual_hiddens: 256 + embedding_dim: 64 +condition: + text: + enabled: true + input_dim: 1024 + embedding_key: cap_emb + dropout: 0.0 + cfg_scale: 1.0 + attribute: + enabled: false + continuous_dim: 0 + discrete_configs: [] + output_dim: 128 + dropout: 0.0 + label: + enabled: false + num_classes: 10 + output_dim: 64 + fusion: concat + condition_dropout: 0.0 +eval: + n_samples: 10 + metrics: + - dtw + - wape + - ed + - crps + - acd + - sd + - kd + - mdd + - fid + - sfd + - jftsd + - prdc_f1 + - joint_prdc_f1 + - cttp + - disc_auc + batch_size: 128 + save_samples: true + sampler: ddim + display_interval: 10 + clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-m/model_configs.yaml + clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-m/clip_model_best.pth + cache_folder: __CACHE_ROOT__/t2s/synth-m/seed0 + use_longalign: false + reference_split: train + disc_auc: + k_folds: 5 + epochs: 5 + batch_size: 128 + lr: 0.001 + weight_decay: 0.0001 + hidden_channels: 128 + num_layers: 3 + dropout: 0.1 + seed: 42 + num_workers: 0 + prdc: + k: 5 + max_samples: 5000 + seed: 0 + metric: euclidean + backend: auto + knn_chunk_size: 4096 + joint_enable: true + joint_weights: + ts: 1.0 + text: 1.0 + joint_normalize: standard + segment_classifier: + enable: false + segment_len: 128 + n_segments: 3 + seed: 42 + acd_max_lag: 50 + mdd_bins: 32 + use_cache: false + cache_file: predictions_cache.pkl + viz: + enable: true + k_cases: 20 + max_vars: 8 + ncols: 5 + seed: 42 + dpi: 200 + alpha: 0.25 + figsize_per_subplot: + - 3.0 + - 2.0 + output_type: pdf +output_dir: '' +created_at: '2025-12-27T23:18:57.418170' diff --git a/experiments/t2s/synth-m/seed0/results/expected_seed_metrics.json b/experiments/t2s/synth-m/seed0/results/expected_seed_metrics.json new file mode 100644 index 0000000000000000000000000000000000000000..4fd1e8eed55cec5cc4aff0152fa06763259a807b --- /dev/null +++ b/experiments/t2s/synth-m/seed0/results/expected_seed_metrics.json @@ -0,0 +1,21 @@ +{ + "dtw": 30.853883330881942, + "wape": 231.39017954533955, + "ed": 36.92706576538086, + "crps": 1.2909548534012474, + "acd": 0.08280777186155319, + "sd": 0.43578305680359264, + "kd": 1.6891005246901758, + "mdd": 0.027735107421874995, + "fid": 122.14499871033982, + "sfd": 317.9769143235806, + "jftsd": 131.62202535095628, + "prdc_f1.f1": 0.0, + "prdc_f1.precision": 0.3175, + "prdc_f1.recall": 0.0, + "joint_prdc_f1.f1": 0.012721153846153844, + "joint_prdc_f1.precision": 0.11025, + "joint_prdc_f1.recall": 0.00675, + "cttp": -2.1494174547195435, + "disc_auc": 1.0 +} \ No newline at end of file diff --git a/experiments/t2s/synth-m/seed0/summary.json b/experiments/t2s/synth-m/seed0/summary.json new file mode 100644 index 0000000000000000000000000000000000000000..e11dcef29063b3f2b84c665d0b2d8a359bb89d35 --- /dev/null +++ b/experiments/t2s/synth-m/seed0/summary.json @@ -0,0 +1,5 @@ +{ + "status": "completed", + "finished_at": "2025-12-28T00:07:42.509933", + "best_checkpoint": "checkpoints/finetune/best.ckpt" +} \ No newline at end of file diff --git a/experiments/t2s/synth-m/seed1/config.template.yaml b/experiments/t2s/synth-m/seed1/config.template.yaml new file mode 100644 index 0000000000000000000000000000000000000000..0239296fb41c4e42ba39d162433bd1677c351a03 --- /dev/null +++ b/experiments/t2s/synth-m/seed1/config.template.yaml @@ -0,0 +1,153 @@ +name: synth-m_t2s +description: '' +seed: 1 +device: cuda:0 +train: + stages: + - name: ae_pretrain + epochs: 200 + lr: 0.001 + use_condition: false + freeze_modules: [] + early_stopping_patience: 50 + - name: finetune + epochs: 500 + lr: 0.001 + use_condition: true + freeze_modules: [] + load_from_stage: ae_pretrain + early_stopping_patience: 50 + epochs: 700 + batch_size: 256 + lr: 0.001 + weight_decay: 0.0001 + scheduler: cosine + scheduler_params: {} + early_stopping_patience: 50 + gradient_clip_val: 1.0 + accumulate_grad_batches: 1 + val_check_interval: 1.0 + num_workers: 4 + pin_memory: true + limit_train_batches: 1.0 + limit_val_batches: 1.0 + limit_test_batches: 1.0 + num_sanity_val_steps: 2 + log_grad_norm: true + log_param_norm: true + log_norm_every_n_steps: 50 +data: + name: synth-m + data_folder: __DATA_ROOT__/datasets/synth-m + n_var: 2 + seq_length: 128 + normalize: false + train_split: 0.8 + val_split: 0.1 +model: + name: t2s + channels: 64 + layers: 4 + nheads: 4 + dropout: 0.1 + patch_size: 2 + mlp_ratio: 2.0 + flow_steps: 10 + cfg_scale: 9.0 + ae: + num_hiddens: 128 + num_residual_layers: 2 + num_residual_hiddens: 256 + embedding_dim: 64 +condition: + text: + enabled: true + input_dim: 1024 + embedding_key: cap_emb + dropout: 0.0 + cfg_scale: 1.0 + attribute: + enabled: false + continuous_dim: 0 + discrete_configs: [] + output_dim: 128 + dropout: 0.0 + label: + enabled: false + num_classes: 10 + output_dim: 64 + fusion: concat + condition_dropout: 0.0 +eval: + n_samples: 10 + metrics: + - dtw + - wape + - ed + - crps + - acd + - sd + - kd + - mdd + - fid + - sfd + - jftsd + - prdc_f1 + - joint_prdc_f1 + - cttp + - disc_auc + batch_size: 128 + save_samples: true + sampler: ddim + display_interval: 10 + clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-m/model_configs.yaml + clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-m/clip_model_best.pth + cache_folder: __CACHE_ROOT__/t2s/synth-m/seed1 + use_longalign: false + reference_split: train + disc_auc: + k_folds: 5 + epochs: 5 + batch_size: 128 + lr: 0.001 + weight_decay: 0.0001 + hidden_channels: 128 + num_layers: 3 + dropout: 0.1 + seed: 42 + num_workers: 0 + prdc: + k: 5 + max_samples: 5000 + seed: 0 + metric: euclidean + backend: auto + knn_chunk_size: 4096 + joint_enable: true + joint_weights: + ts: 1.0 + text: 1.0 + joint_normalize: standard + segment_classifier: + enable: false + segment_len: 128 + n_segments: 3 + seed: 42 + acd_max_lag: 50 + mdd_bins: 32 + use_cache: false + cache_file: predictions_cache.pkl + viz: + enable: true + k_cases: 20 + max_vars: 8 + ncols: 5 + seed: 42 + dpi: 200 + alpha: 0.25 + figsize_per_subplot: + - 3.0 + - 2.0 + output_type: pdf +output_dir: '' +created_at: '2025-12-27T23:29:31.014346' diff --git a/experiments/t2s/synth-m/seed1/results/expected_seed_metrics.json b/experiments/t2s/synth-m/seed1/results/expected_seed_metrics.json new file mode 100644 index 0000000000000000000000000000000000000000..cc7f24c20b95b6a3f910ae324494432a461e7656 --- /dev/null +++ b/experiments/t2s/synth-m/seed1/results/expected_seed_metrics.json @@ -0,0 +1,21 @@ +{ + "dtw": 51.97795645532827, + "wape": 375.2316361826495, + "ed": 57.725555297851564, + "crps": 2.47075430506335, + "acd": 0.036171168088912964, + "sd": 0.48953894824419564, + "kd": 1.4351366797985663, + "mdd": 0.03595974731445312, + "fid": 109.6813469453873, + "sfd": 409.9287758892757, + "jftsd": 118.27890977769226, + "prdc_f1.f1": 0.0, + "prdc_f1.precision": 0.4685, + "prdc_f1.recall": 0.0, + "joint_prdc_f1.f1": 0.02031169665809768, + "joint_prdc_f1.precision": 0.18375, + "joint_prdc_f1.recall": 0.01075, + "cttp": 4.7206211853027344, + "disc_auc": 1.0 +} \ No newline at end of file diff --git a/experiments/t2s/synth-m/seed1/summary.json b/experiments/t2s/synth-m/seed1/summary.json new file mode 100644 index 0000000000000000000000000000000000000000..d7f1e76d8ebab651f7b94350b8b9d4f8ffff0e59 --- /dev/null +++ b/experiments/t2s/synth-m/seed1/summary.json @@ -0,0 +1,5 @@ +{ + "status": "completed", + "finished_at": "2025-12-28T00:09:22.363900", + "best_checkpoint": "checkpoints/finetune/best.ckpt" +} \ No newline at end of file diff --git a/experiments/t2s/synth-m/seed2/config.template.yaml b/experiments/t2s/synth-m/seed2/config.template.yaml new file mode 100644 index 0000000000000000000000000000000000000000..b4f422d79f1434b323954b7410d6c1f6fae49b79 --- /dev/null +++ b/experiments/t2s/synth-m/seed2/config.template.yaml @@ -0,0 +1,153 @@ +name: synth-m_t2s +description: '' +seed: 2 +device: cuda:0 +train: + stages: + - name: ae_pretrain + epochs: 200 + lr: 0.001 + use_condition: false + freeze_modules: [] + early_stopping_patience: 50 + - name: finetune + epochs: 500 + lr: 0.001 + use_condition: true + freeze_modules: [] + load_from_stage: ae_pretrain + early_stopping_patience: 50 + epochs: 700 + batch_size: 256 + lr: 0.001 + weight_decay: 0.0001 + scheduler: cosine + scheduler_params: {} + early_stopping_patience: 50 + gradient_clip_val: 1.0 + accumulate_grad_batches: 1 + val_check_interval: 1.0 + num_workers: 4 + pin_memory: true + limit_train_batches: 1.0 + limit_val_batches: 1.0 + limit_test_batches: 1.0 + num_sanity_val_steps: 2 + log_grad_norm: true + log_param_norm: true + log_norm_every_n_steps: 50 +data: + name: synth-m + data_folder: __DATA_ROOT__/datasets/synth-m + n_var: 2 + seq_length: 128 + normalize: false + train_split: 0.8 + val_split: 0.1 +model: + name: t2s + channels: 64 + layers: 4 + nheads: 4 + dropout: 0.1 + patch_size: 2 + mlp_ratio: 2.0 + flow_steps: 10 + cfg_scale: 9.0 + ae: + num_hiddens: 128 + num_residual_layers: 2 + num_residual_hiddens: 256 + embedding_dim: 64 +condition: + text: + enabled: true + input_dim: 1024 + embedding_key: cap_emb + dropout: 0.0 + cfg_scale: 1.0 + attribute: + enabled: false + continuous_dim: 0 + discrete_configs: [] + output_dim: 128 + dropout: 0.0 + label: + enabled: false + num_classes: 10 + output_dim: 64 + fusion: concat + condition_dropout: 0.0 +eval: + n_samples: 10 + metrics: + - dtw + - wape + - ed + - crps + - acd + - sd + - kd + - mdd + - fid + - sfd + - jftsd + - prdc_f1 + - joint_prdc_f1 + - cttp + - disc_auc + batch_size: 128 + save_samples: true + sampler: ddim + display_interval: 10 + clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-m/model_configs.yaml + clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-m/clip_model_best.pth + cache_folder: __CACHE_ROOT__/t2s/synth-m/seed2 + use_longalign: false + reference_split: train + disc_auc: + k_folds: 5 + epochs: 5 + batch_size: 128 + lr: 0.001 + weight_decay: 0.0001 + hidden_channels: 128 + num_layers: 3 + dropout: 0.1 + seed: 42 + num_workers: 0 + prdc: + k: 5 + max_samples: 5000 + seed: 0 + metric: euclidean + backend: auto + knn_chunk_size: 4096 + joint_enable: true + joint_weights: + ts: 1.0 + text: 1.0 + joint_normalize: standard + segment_classifier: + enable: false + segment_len: 128 + n_segments: 3 + seed: 42 + acd_max_lag: 50 + mdd_bins: 32 + use_cache: false + cache_file: predictions_cache.pkl + viz: + enable: true + k_cases: 20 + max_vars: 8 + ncols: 5 + seed: 42 + dpi: 200 + alpha: 0.25 + figsize_per_subplot: + - 3.0 + - 2.0 + output_type: pdf +output_dir: '' +created_at: '2025-12-27T23:36:31.656409' diff --git a/experiments/t2s/synth-m/seed2/results/expected_seed_metrics.json b/experiments/t2s/synth-m/seed2/results/expected_seed_metrics.json new file mode 100644 index 0000000000000000000000000000000000000000..79a081493e78efa237470741322841e6f36b2139 --- /dev/null +++ b/experiments/t2s/synth-m/seed2/results/expected_seed_metrics.json @@ -0,0 +1,21 @@ +{ + "dtw": 33.4548552772021, + "wape": 233.52481543411972, + "ed": 38.108337890625, + "crps": 1.2779978524073943, + "acd": 0.10699678212404251, + "sd": 0.26627323448919915, + "kd": 1.3601897974690218, + "mdd": 0.027756306966145835, + "fid": 109.19084633158991, + "sfd": 384.584061896183, + "jftsd": 119.15697446699316, + "prdc_f1.f1": 0.0, + "prdc_f1.precision": 0.48775, + "prdc_f1.recall": 0.0, + "joint_prdc_f1.f1": 0.02104734576757532, + "joint_prdc_f1.precision": 0.163, + "joint_prdc_f1.recall": 0.01125, + "cttp": 2.3761753463745117, + "disc_auc": 1.0 +} \ No newline at end of file diff --git a/experiments/t2s/synth-m/seed2/summary.json b/experiments/t2s/synth-m/seed2/summary.json new file mode 100644 index 0000000000000000000000000000000000000000..b237776ba988a0e1e863312d7fd2c415440656d5 --- /dev/null +++ b/experiments/t2s/synth-m/seed2/summary.json @@ -0,0 +1,5 @@ +{ + "status": "completed", + "finished_at": "2025-12-28T00:23:21.692223", + "best_checkpoint": "checkpoints/finetune/best.ckpt" +} \ No newline at end of file diff --git a/experiments/t2s/synth-u/seed0/config.template.yaml b/experiments/t2s/synth-u/seed0/config.template.yaml new file mode 100644 index 0000000000000000000000000000000000000000..7746ac9cc736bc4866805d8555318b233d5b9d17 --- /dev/null +++ b/experiments/t2s/synth-u/seed0/config.template.yaml @@ -0,0 +1,153 @@ +name: synth-u_t2s +description: '' +seed: 0 +device: cuda:0 +train: + stages: + - name: ae_pretrain + epochs: 200 + lr: 0.001 + use_condition: false + freeze_modules: [] + early_stopping_patience: 50 + - name: finetune + epochs: 500 + lr: 0.001 + use_condition: true + freeze_modules: [] + load_from_stage: ae_pretrain + early_stopping_patience: 50 + epochs: 700 + batch_size: 256 + lr: 0.001 + weight_decay: 0.0001 + scheduler: cosine + scheduler_params: {} + early_stopping_patience: 50 + gradient_clip_val: 1.0 + accumulate_grad_batches: 1 + val_check_interval: 1.0 + num_workers: 4 + pin_memory: true + limit_train_batches: 1.0 + limit_val_batches: 1.0 + limit_test_batches: 1.0 + num_sanity_val_steps: 2 + log_grad_norm: true + log_param_norm: true + log_norm_every_n_steps: 50 +data: + name: synth-u + data_folder: __DATA_ROOT__/datasets/synth-u + n_var: 1 + seq_length: 128 + normalize: false + train_split: 0.8 + val_split: 0.1 +model: + name: t2s + channels: 64 + layers: 4 + nheads: 4 + dropout: 0.1 + patch_size: 2 + mlp_ratio: 2.0 + flow_steps: 10 + cfg_scale: 9.0 + ae: + num_hiddens: 128 + num_residual_layers: 2 + num_residual_hiddens: 256 + embedding_dim: 64 +condition: + text: + enabled: true + input_dim: 1024 + embedding_key: cap_emb + dropout: 0.0 + cfg_scale: 1.0 + attribute: + enabled: false + continuous_dim: 0 + discrete_configs: [] + output_dim: 128 + dropout: 0.0 + label: + enabled: false + num_classes: 10 + output_dim: 64 + fusion: concat + condition_dropout: 0.0 +eval: + n_samples: 10 + metrics: + - dtw + - wape + - ed + - crps + - acd + - sd + - kd + - mdd + - fid + - sfd + - jftsd + - prdc_f1 + - joint_prdc_f1 + - cttp + - disc_auc + batch_size: 128 + save_samples: true + sampler: ddim + display_interval: 10 + clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-u/model_configs.yaml + clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-u/clip_model_best.pth + cache_folder: __CACHE_ROOT__/t2s/synth-u/seed0 + use_longalign: false + reference_split: train + disc_auc: + k_folds: 5 + epochs: 5 + batch_size: 128 + lr: 0.001 + weight_decay: 0.0001 + hidden_channels: 128 + num_layers: 3 + dropout: 0.1 + seed: 42 + num_workers: 0 + prdc: + k: 5 + max_samples: 5000 + seed: 0 + metric: euclidean + backend: auto + knn_chunk_size: 4096 + joint_enable: true + joint_weights: + ts: 1.0 + text: 1.0 + joint_normalize: standard + segment_classifier: + enable: false + segment_len: 128 + n_segments: 3 + seed: 42 + acd_max_lag: 50 + mdd_bins: 32 + use_cache: false + cache_file: predictions_cache.pkl + viz: + enable: true + k_cases: 20 + max_vars: 8 + ncols: 5 + seed: 42 + dpi: 200 + alpha: 0.25 + figsize_per_subplot: + - 3.0 + - 2.0 + output_type: pdf +output_dir: '' +created_at: '2025-12-27T23:40:52.259276' diff --git a/experiments/t2s/synth-u/seed0/results/expected_seed_metrics.json b/experiments/t2s/synth-u/seed0/results/expected_seed_metrics.json new file mode 100644 index 0000000000000000000000000000000000000000..6393e380f5c9c430e9fb4861ed2defa503698f46 --- /dev/null +++ b/experiments/t2s/synth-u/seed0/results/expected_seed_metrics.json @@ -0,0 +1,21 @@ +{ + "dtw": 47.12652707117338, + "wape": 550.2959810862212, + "ed": 54.415912963867186, + "crps": 3.4977831919962763, + "acd": 0.039769306778907776, + "sd": 0.32305185133674863, + "kd": 0.5842860290533332, + "mdd": 0.05455151367187498, + "fid": 175.4499695138689, + "sfd": 158.77707966819406, + "jftsd": 186.9348911336512, + "prdc_f1.f1": 0.0, + "prdc_f1.precision": 0.01575, + "prdc_f1.recall": 0.0, + "joint_prdc_f1.f1": 0.005913978494623656, + "joint_prdc_f1.precision": 0.20625, + "joint_prdc_f1.recall": 0.003, + "cttp": 5.37317338180542, + "disc_auc": 1.0 +} \ No newline at end of file diff --git a/experiments/t2s/synth-u/seed0/summary.json b/experiments/t2s/synth-u/seed0/summary.json new file mode 100644 index 0000000000000000000000000000000000000000..405b277f2321ecc604afe10dcc1a23e52ccbf9a5 --- /dev/null +++ b/experiments/t2s/synth-u/seed0/summary.json @@ -0,0 +1,5 @@ +{ + "status": "completed", + "finished_at": "2025-12-28T00:15:00.860457", + "best_checkpoint": "checkpoints/finetune/best.ckpt" +} \ No newline at end of file diff --git a/experiments/t2s/synth-u/seed1/config.template.yaml b/experiments/t2s/synth-u/seed1/config.template.yaml new file mode 100644 index 0000000000000000000000000000000000000000..85154b0c399d2ed7328e92d4f40ebbf07c6f14e9 --- /dev/null +++ b/experiments/t2s/synth-u/seed1/config.template.yaml @@ -0,0 +1,153 @@ +name: synth-u_t2s +description: '' +seed: 1 +device: cuda:0 +train: + stages: + - name: ae_pretrain + epochs: 200 + lr: 0.001 + use_condition: false + freeze_modules: [] + early_stopping_patience: 50 + - name: finetune + epochs: 500 + lr: 0.001 + use_condition: true + freeze_modules: [] + load_from_stage: ae_pretrain + early_stopping_patience: 50 + epochs: 700 + batch_size: 256 + lr: 0.001 + weight_decay: 0.0001 + scheduler: cosine + scheduler_params: {} + early_stopping_patience: 50 + gradient_clip_val: 1.0 + accumulate_grad_batches: 1 + val_check_interval: 1.0 + num_workers: 4 + pin_memory: true + limit_train_batches: 1.0 + limit_val_batches: 1.0 + limit_test_batches: 1.0 + num_sanity_val_steps: 2 + log_grad_norm: true + log_param_norm: true + log_norm_every_n_steps: 50 +data: + name: synth-u + data_folder: __DATA_ROOT__/datasets/synth-u + n_var: 1 + seq_length: 128 + normalize: false + train_split: 0.8 + val_split: 0.1 +model: + name: t2s + channels: 64 + layers: 4 + nheads: 4 + dropout: 0.1 + patch_size: 2 + mlp_ratio: 2.0 + flow_steps: 10 + cfg_scale: 9.0 + ae: + num_hiddens: 128 + num_residual_layers: 2 + num_residual_hiddens: 256 + embedding_dim: 64 +condition: + text: + enabled: true + input_dim: 1024 + embedding_key: cap_emb + dropout: 0.0 + cfg_scale: 1.0 + attribute: + enabled: false + continuous_dim: 0 + discrete_configs: [] + output_dim: 128 + dropout: 0.0 + label: + enabled: false + num_classes: 10 + output_dim: 64 + fusion: concat + condition_dropout: 0.0 +eval: + n_samples: 10 + metrics: + - dtw + - wape + - ed + - crps + - acd + - sd + - kd + - mdd + - fid + - sfd + - jftsd + - prdc_f1 + - joint_prdc_f1 + - cttp + - disc_auc + batch_size: 128 + save_samples: true + sampler: ddim + display_interval: 10 + clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-u/model_configs.yaml + clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-u/clip_model_best.pth + cache_folder: __CACHE_ROOT__/t2s/synth-u/seed1 + use_longalign: false + reference_split: train + disc_auc: + k_folds: 5 + epochs: 5 + batch_size: 128 + lr: 0.001 + weight_decay: 0.0001 + hidden_channels: 128 + num_layers: 3 + dropout: 0.1 + seed: 42 + num_workers: 0 + prdc: + k: 5 + max_samples: 5000 + seed: 0 + metric: euclidean + backend: auto + knn_chunk_size: 4096 + joint_enable: true + joint_weights: + ts: 1.0 + text: 1.0 + joint_normalize: standard + segment_classifier: + enable: false + segment_len: 128 + n_segments: 3 + seed: 42 + acd_max_lag: 50 + mdd_bins: 32 + use_cache: false + cache_file: predictions_cache.pkl + viz: + enable: true + k_cases: 20 + max_vars: 8 + ncols: 5 + seed: 42 + dpi: 200 + alpha: 0.25 + figsize_per_subplot: + - 3.0 + - 2.0 + output_type: pdf +output_dir: '' +created_at: '2025-12-27T23:41:04.392813' diff --git a/experiments/t2s/synth-u/seed1/results/expected_seed_metrics.json b/experiments/t2s/synth-u/seed1/results/expected_seed_metrics.json new file mode 100644 index 0000000000000000000000000000000000000000..48fbf08de88fc5aedf5abf709e37a185a00dbc55 --- /dev/null +++ b/experiments/t2s/synth-u/seed1/results/expected_seed_metrics.json @@ -0,0 +1,21 @@ +{ + "dtw": 15.365408773958965, + "wape": 178.296819353186, + "ed": 20.013489288330078, + "crps": 1.0322840045545667, + "acd": 0.04774828627705574, + "sd": 0.9647802643913168, + "kd": 1.2767695148803266, + "mdd": 0.02350620524088541, + "fid": 134.2823978837169, + "sfd": 79.6028833155586, + "jftsd": 147.3762497463016, + "prdc_f1.f1": 0.0, + "prdc_f1.precision": 0.4055, + "prdc_f1.recall": 0.0, + "joint_prdc_f1.f1": 0.027200400000000003, + "joint_prdc_f1.precision": 0.29825, + "joint_prdc_f1.recall": 0.01425, + "cttp": 12.133055778503419, + "disc_auc": 1.0 +} \ No newline at end of file diff --git a/experiments/t2s/synth-u/seed1/summary.json b/experiments/t2s/synth-u/seed1/summary.json new file mode 100644 index 0000000000000000000000000000000000000000..51647b53e342a6530c83dfd7e61621464914eb07 --- /dev/null +++ b/experiments/t2s/synth-u/seed1/summary.json @@ -0,0 +1,5 @@ +{ + "status": "completed", + "finished_at": "2025-12-28T00:25:13.017144", + "best_checkpoint": "checkpoints/finetune/best.ckpt" +} \ No newline at end of file diff --git a/experiments/t2s/synth-u/seed2/config.template.yaml b/experiments/t2s/synth-u/seed2/config.template.yaml new file mode 100644 index 0000000000000000000000000000000000000000..c2d533be9983609e035f880ffb87ac590c89d738 --- /dev/null +++ b/experiments/t2s/synth-u/seed2/config.template.yaml @@ -0,0 +1,153 @@ +name: synth-u_t2s +description: '' +seed: 2 +device: cuda:0 +train: + stages: + - name: ae_pretrain + epochs: 200 + lr: 0.001 + use_condition: false + freeze_modules: [] + early_stopping_patience: 50 + - name: finetune + epochs: 500 + lr: 0.001 + use_condition: true + freeze_modules: [] + load_from_stage: ae_pretrain + early_stopping_patience: 50 + epochs: 700 + batch_size: 256 + lr: 0.001 + weight_decay: 0.0001 + scheduler: cosine + scheduler_params: {} + early_stopping_patience: 50 + gradient_clip_val: 1.0 + accumulate_grad_batches: 1 + val_check_interval: 1.0 + num_workers: 4 + pin_memory: true + limit_train_batches: 1.0 + limit_val_batches: 1.0 + limit_test_batches: 1.0 + num_sanity_val_steps: 2 + log_grad_norm: true + log_param_norm: true + log_norm_every_n_steps: 50 +data: + name: synth-u + data_folder: __DATA_ROOT__/datasets/synth-u + n_var: 1 + seq_length: 128 + normalize: false + train_split: 0.8 + val_split: 0.1 +model: + name: t2s + channels: 64 + layers: 4 + nheads: 4 + dropout: 0.1 + patch_size: 2 + mlp_ratio: 2.0 + flow_steps: 10 + cfg_scale: 9.0 + ae: + num_hiddens: 128 + num_residual_layers: 2 + num_residual_hiddens: 256 + embedding_dim: 64 +condition: + text: + enabled: true + input_dim: 1024 + embedding_key: cap_emb + dropout: 0.0 + cfg_scale: 1.0 + attribute: + enabled: false + continuous_dim: 0 + discrete_configs: [] + output_dim: 128 + dropout: 0.0 + label: + enabled: false + num_classes: 10 + output_dim: 64 + fusion: concat + condition_dropout: 0.0 +eval: + n_samples: 10 + metrics: + - dtw + - wape + - ed + - crps + - acd + - sd + - kd + - mdd + - fid + - sfd + - jftsd + - prdc_f1 + - joint_prdc_f1 + - cttp + - disc_auc + batch_size: 128 + save_samples: true + sampler: ddim + display_interval: 10 + clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-u/model_configs.yaml + clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-u/clip_model_best.pth + cache_folder: __CACHE_ROOT__/t2s/synth-u/seed2 + use_longalign: false + reference_split: train + disc_auc: + k_folds: 5 + epochs: 5 + batch_size: 128 + lr: 0.001 + weight_decay: 0.0001 + hidden_channels: 128 + num_layers: 3 + dropout: 0.1 + seed: 42 + num_workers: 0 + prdc: + k: 5 + max_samples: 5000 + seed: 0 + metric: euclidean + backend: auto + knn_chunk_size: 4096 + joint_enable: true + joint_weights: + ts: 1.0 + text: 1.0 + joint_normalize: standard + segment_classifier: + enable: false + segment_len: 128 + n_segments: 3 + seed: 42 + acd_max_lag: 50 + mdd_bins: 32 + use_cache: false + cache_file: predictions_cache.pkl + viz: + enable: true + k_cases: 20 + max_vars: 8 + ncols: 5 + seed: 42 + dpi: 200 + alpha: 0.25 + figsize_per_subplot: + - 3.0 + - 2.0 + output_type: pdf +output_dir: '' +created_at: '2025-12-27T23:48:23.773306' diff --git a/experiments/t2s/synth-u/seed2/results/expected_seed_metrics.json b/experiments/t2s/synth-u/seed2/results/expected_seed_metrics.json new file mode 100644 index 0000000000000000000000000000000000000000..00fa40a3cc86b7dd35cdc976c63e53328d4851a2 --- /dev/null +++ b/experiments/t2s/synth-u/seed2/results/expected_seed_metrics.json @@ -0,0 +1,21 @@ +{ + "dtw": 19.35132308105883, + "wape": 216.31664117981865, + "ed": 23.888965423583983, + "crps": 1.2506499665516893, + "acd": 0.05925827473402023, + "sd": 0.20025774443618977, + "kd": 0.4887130778990958, + "mdd": 0.029227783203124996, + "fid": 123.7158536108432, + "sfd": 102.84906228929884, + "jftsd": 134.65165898637255, + "prdc_f1.f1": 0.0, + "prdc_f1.precision": 0.27775, + "prdc_f1.recall": 0.0, + "joint_prdc_f1.f1": 0.046265346534653466, + "joint_prdc_f1.precision": 0.354, + "joint_prdc_f1.recall": 0.02475, + "cttp": 13.46905096435547, + "disc_auc": 1.0 +} \ No newline at end of file diff --git a/experiments/t2s/synth-u/seed2/summary.json b/experiments/t2s/synth-u/seed2/summary.json new file mode 100644 index 0000000000000000000000000000000000000000..70f0f8768b85e16c7c2f6565eea72cf8fb5e9a5a --- /dev/null +++ b/experiments/t2s/synth-u/seed2/summary.json @@ -0,0 +1,5 @@ +{ + "status": "completed", + "finished_at": "2025-12-28T00:23:34.396797", + "best_checkpoint": "checkpoints/finetune/best.ckpt" +} \ No newline at end of file diff --git a/experiments/tedit/synth-m/seed0/checkpoints/finetune/best.ckpt b/experiments/tedit/synth-m/seed0/checkpoints/finetune/best.ckpt new file mode 100644 index 0000000000000000000000000000000000000000..6d49dcebc3513aa2ad73281509bf6b442c8fb2d1 --- /dev/null +++ b/experiments/tedit/synth-m/seed0/checkpoints/finetune/best.ckpt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c5bcd1bbf275b90bdb268ee73dfb931e0d724407415ba42694704511bae2a811 +size 14758842 diff --git a/experiments/tedit/synth-m/seed1/checkpoints/finetune/best.ckpt b/experiments/tedit/synth-m/seed1/checkpoints/finetune/best.ckpt new file mode 100644 index 0000000000000000000000000000000000000000..23d5a555a5041e145f0074154b7d3f581913f7c9 --- /dev/null +++ b/experiments/tedit/synth-m/seed1/checkpoints/finetune/best.ckpt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7476a4ebf9e19fc690881059a95ea379eea286c22c995abc90247ce57bd2de79 +size 14758842 diff --git a/experiments/tedit/synth-u/seed0/checkpoints/finetune/best.ckpt b/experiments/tedit/synth-u/seed0/checkpoints/finetune/best.ckpt new file mode 100644 index 0000000000000000000000000000000000000000..cd7b7ce0d44d43f9723befe947ccb9c4feefe3a9 --- /dev/null +++ b/experiments/tedit/synth-u/seed0/checkpoints/finetune/best.ckpt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:070094289a942e1704531ea00f897aaeeb913de59e3d9f50dbf8dfea869a0e03 +size 14117458 diff --git a/experiments/tedit/synth-u/seed1/checkpoints/finetune/best.ckpt b/experiments/tedit/synth-u/seed1/checkpoints/finetune/best.ckpt new file mode 100644 index 0000000000000000000000000000000000000000..4bb8dc309b7c47dd0eca40db8faff6201117e47c --- /dev/null +++ b/experiments/tedit/synth-u/seed1/checkpoints/finetune/best.ckpt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:051d8050c3587f681ed8be44aeb296096969b7c2e3b4a78365a8950ebe473599 +size 14117458 diff --git a/experiments/tedit/synth-u/seed2/checkpoints/finetune/best.ckpt b/experiments/tedit/synth-u/seed2/checkpoints/finetune/best.ckpt new file mode 100644 index 0000000000000000000000000000000000000000..7d31260b2c6508debea20c51530dfac0dc8b7b1b --- /dev/null +++ b/experiments/tedit/synth-u/seed2/checkpoints/finetune/best.ckpt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b63fb74d011330cfc938770fd82f3339c3dccf95fffcf7d7f7fbdb3fc595e691 +size 14117458 diff --git a/experiments/timevqvae/synth-m/seed0/checkpoints/finetune/best.ckpt b/experiments/timevqvae/synth-m/seed0/checkpoints/finetune/best.ckpt new file mode 100644 index 0000000000000000000000000000000000000000..a2fc3530b68fdfa1f8530e72c61f90d9e994009f --- /dev/null +++ b/experiments/timevqvae/synth-m/seed0/checkpoints/finetune/best.ckpt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:17d74607d5b60a6d9d6c1ada8a683b879878b7481e359708c6f000c9a8a87207 +size 24656112 diff --git a/experiments/timevqvae/synth-m/seed1/checkpoints/finetune/best.ckpt b/experiments/timevqvae/synth-m/seed1/checkpoints/finetune/best.ckpt new file mode 100644 index 0000000000000000000000000000000000000000..fe9f0efcf7c84c0fb5bb658b9e4f96d067a85f3e --- /dev/null +++ b/experiments/timevqvae/synth-m/seed1/checkpoints/finetune/best.ckpt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:894e62d7099959245a1a9ed2913a5f1a3936ac68a760603d17cba9266cb474fa +size 24656112 diff --git a/experiments/timevqvae/synth-m/seed2/checkpoints/finetune/best.ckpt b/experiments/timevqvae/synth-m/seed2/checkpoints/finetune/best.ckpt new file mode 100644 index 0000000000000000000000000000000000000000..c4063171f4707fc1f01b79a16eb83b89673247f7 --- /dev/null +++ b/experiments/timevqvae/synth-m/seed2/checkpoints/finetune/best.ckpt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:76215384630b0827c6489c2d85c1cbe78fda7eb16c637bfa89ee53c20abe4439 +size 24656112 diff --git a/experiments/timevqvae/synth-u/seed0/checkpoints/finetune/best.ckpt b/experiments/timevqvae/synth-u/seed0/checkpoints/finetune/best.ckpt new file mode 100644 index 0000000000000000000000000000000000000000..0d2984bff54f44cabd147ccc0056a35434bbb7a4 --- /dev/null +++ b/experiments/timevqvae/synth-u/seed0/checkpoints/finetune/best.ckpt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b8785b1798bc972bae7fe72e0b054a85a6236aa7887a76d72f2b645840410161 +size 24212336 diff --git a/experiments/timevqvae/synth-u/seed1/checkpoints/finetune/best.ckpt b/experiments/timevqvae/synth-u/seed1/checkpoints/finetune/best.ckpt new file mode 100644 index 0000000000000000000000000000000000000000..fb30ebbc7c014301fdf0b131bd5c1ac246d25439 --- /dev/null +++ b/experiments/timevqvae/synth-u/seed1/checkpoints/finetune/best.ckpt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c7241060710e3b4beb3d7c3a3c13781daf2dd7b7cba1065cb0fd0210c977f451 +size 24212336 diff --git a/experiments/timevqvae/synth-u/seed2/checkpoints/finetune/best.ckpt b/experiments/timevqvae/synth-u/seed2/checkpoints/finetune/best.ckpt new file mode 100644 index 0000000000000000000000000000000000000000..af468895da43fb823b14e18f045ad97071233708 --- /dev/null +++ b/experiments/timevqvae/synth-u/seed2/checkpoints/finetune/best.ckpt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4b4fdfa5ca5b59dc0193bcf41845c0c99d6d938ff11777bfc113b76e46bc0a7b +size 24212336 diff --git a/experiments/timeweaver/synth-m/seed0/results/expected_seed_metrics.json b/experiments/timeweaver/synth-m/seed0/results/expected_seed_metrics.json new file mode 100644 index 0000000000000000000000000000000000000000..0b79635cf5b478c06d871122708a1c8eb8a2f516 --- /dev/null +++ b/experiments/timeweaver/synth-m/seed0/results/expected_seed_metrics.json @@ -0,0 +1,21 @@ +{ + "dtw": 9.638331482196921, + "wape": 87.69841286522364, + "ed": 12.988000579833985, + "crps": 0.443040543349335, + "acd": 0.05639765411615372, + "sd": 0.032235342662713815, + "kd": 0.3527206720753482, + "mdd": 0.01511520385742187, + "fid": 29.344011810966066, + "sfd": 27.191097103191552, + "jftsd": 38.20762139590508, + "prdc_f1.f1": 0.40610068093385215, + "prdc_f1.precision": 0.7495, + "prdc_f1.recall": 0.2785, + "joint_prdc_f1.f1": 0.5496895098691505, + "joint_prdc_f1.precision": 0.65225, + "joint_prdc_f1.recall": 0.475, + "cttp": 21.776594619750977, + "disc_auc": 0.99048296875 +} \ No newline at end of file diff --git a/experiments/timeweaver/synth-m/seed1/config.template.yaml b/experiments/timeweaver/synth-m/seed1/config.template.yaml new file mode 100644 index 0000000000000000000000000000000000000000..0a8fa0817c8232353f9995eca9b589d50ae6664f --- /dev/null +++ b/experiments/timeweaver/synth-m/seed1/config.template.yaml @@ -0,0 +1,155 @@ +name: synth-m_timeweaver +description: '' +seed: 1 +device: cuda:0 +train: + stages: + - name: finetune + epochs: 700 + lr: 0.001 + use_condition: true + freeze_modules: [] + early_stopping_patience: 50 + epochs: 700 + batch_size: 256 + lr: 0.001 + weight_decay: 0.0001 + scheduler: cosine + scheduler_params: {} + early_stopping_patience: 50 + gradient_clip_val: 1.0 + accumulate_grad_batches: 1 + val_check_interval: 1.0 + num_workers: 4 + pin_memory: true + limit_train_batches: 1.0 + limit_val_batches: 1.0 + limit_test_batches: 1.0 + num_sanity_val_steps: 2 + log_grad_norm: true + log_param_norm: true + log_norm_every_n_steps: 50 +data: + name: synth-m + data_folder: __DATA_ROOT__/datasets/synth-m + n_var: 2 + seq_length: 128 + normalize: false + train_split: 0.8 + val_split: 0.1 +model: + name: timeweaver + channels: 64 + layers: 3 + nheads: 8 + dropout: 0.1 + attr_dim: 64 + num_attr_heads: 4 + diffusion_steps: 50 + noise_schedule: quad + beta_start: 0.0001 + beta_end: 0.5 + base_patch: 4 + attr_proj_type: avg + attr_heads: 8 + attr_layers: 2 + n_attrs: 4 + feature_ratio: 0.5 + side_var_emb: 16 + side_time_emb: 112 +condition: + text: + enabled: false + input_dim: 1024 + embedding_key: cap_emb + dropout: 0.0 + cfg_scale: 1.0 + attribute: + enabled: true + continuous_dim: 0 + discrete_configs: + - num_classes: 4 + - num_classes: 2 + - num_classes: 4 + - num_classes: 4 + output_dim: 128 + dropout: 0.0 + label: + enabled: false + num_classes: 10 + output_dim: 64 + fusion: concat + condition_dropout: 0.0 +eval: + n_samples: 10 + metrics: + - dtw + - wape + - ed + - crps + - acd + - sd + - kd + - mdd + - fid + - sfd + - jftsd + - prdc_f1 + - joint_prdc_f1 + - cttp + - disc_auc + batch_size: 128 + save_samples: true + sampler: ddim + display_interval: 10 + clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-m/model_configs.yaml + clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-m/clip_model_best.pth + cache_folder: __CACHE_ROOT__/timeweaver/synth-m/seed1 + use_longalign: false + reference_split: train + disc_auc: + k_folds: 5 + epochs: 5 + batch_size: 128 + lr: 0.001 + weight_decay: 0.0001 + hidden_channels: 128 + num_layers: 3 + dropout: 0.1 + seed: 42 + num_workers: 0 + prdc: + k: 5 + max_samples: 5000 + seed: 0 + metric: euclidean + backend: auto + knn_chunk_size: 4096 + joint_enable: true + joint_weights: + ts: 1.0 + text: 1.0 + joint_normalize: standard + segment_classifier: + enable: false + segment_len: 128 + n_segments: 3 + seed: 42 + acd_max_lag: 50 + mdd_bins: 32 + use_cache: false + cache_file: predictions_cache.pkl + viz: + enable: true + k_cases: 20 + max_vars: 8 + ncols: 5 + seed: 42 + dpi: 200 + alpha: 0.25 + figsize_per_subplot: + - 3.0 + - 2.0 + output_type: pdf +output_dir: '' +created_at: '2025-12-27T20:29:41.415684' diff --git a/experiments/timeweaver/synth-m/seed1/results/expected_seed_metrics.json b/experiments/timeweaver/synth-m/seed1/results/expected_seed_metrics.json new file mode 100644 index 0000000000000000000000000000000000000000..9469b4ebd815a451bb115a47c3255ce07fa3a7c4 --- /dev/null +++ b/experiments/timeweaver/synth-m/seed1/results/expected_seed_metrics.json @@ -0,0 +1,21 @@ +{ + "dtw": 10.114984882914769, + "wape": 91.04607251023698, + "ed": 13.448069694519043, + "crps": 0.49989060771452826, + "acd": 0.052001263946294785, + "sd": 0.045910696879056724, + "kd": 0.07278378424386167, + "mdd": 0.014654602050781246, + "fid": 29.582095506289704, + "sfd": 27.51172455169447, + "jftsd": 38.916594517851706, + "prdc_f1.f1": 0.41071530068510537, + "prdc_f1.precision": 0.6935, + "prdc_f1.recall": 0.29175, + "joint_prdc_f1.f1": 0.5048435530780417, + "joint_prdc_f1.precision": 0.59075, + "joint_prdc_f1.recall": 0.44075, + "cttp": 20.934710494995116, + "disc_auc": 0.99168421875 +} \ No newline at end of file diff --git a/experiments/timeweaver/synth-m/seed1/summary.json b/experiments/timeweaver/synth-m/seed1/summary.json new file mode 100644 index 0000000000000000000000000000000000000000..45cd2dbf7a7eb17f5f6389aff2e0202ad16eb415 --- /dev/null +++ b/experiments/timeweaver/synth-m/seed1/summary.json @@ -0,0 +1,5 @@ +{ + "status": "completed", + "finished_at": "2025-12-27T21:03:53.324546", + "best_checkpoint": "checkpoints/finetune/best.ckpt" +} \ No newline at end of file diff --git a/experiments/timeweaver/synth-m/seed2/config.template.yaml b/experiments/timeweaver/synth-m/seed2/config.template.yaml new file mode 100644 index 0000000000000000000000000000000000000000..2cdbb9f21b12d740b8f4ae601e2f3dff0766ccc9 --- /dev/null +++ b/experiments/timeweaver/synth-m/seed2/config.template.yaml @@ -0,0 +1,155 @@ +name: synth-m_timeweaver +description: '' +seed: 2 +device: cuda:0 +train: + stages: + - name: finetune + epochs: 700 + lr: 0.001 + use_condition: true + freeze_modules: [] + early_stopping_patience: 50 + epochs: 700 + batch_size: 256 + lr: 0.001 + weight_decay: 0.0001 + scheduler: cosine + scheduler_params: {} + early_stopping_patience: 50 + gradient_clip_val: 1.0 + accumulate_grad_batches: 1 + val_check_interval: 1.0 + num_workers: 4 + pin_memory: true + limit_train_batches: 1.0 + limit_val_batches: 1.0 + limit_test_batches: 1.0 + num_sanity_val_steps: 2 + log_grad_norm: true + log_param_norm: true + log_norm_every_n_steps: 50 +data: + name: synth-m + data_folder: __DATA_ROOT__/datasets/synth-m + n_var: 2 + seq_length: 128 + normalize: false + train_split: 0.8 + val_split: 0.1 +model: + name: timeweaver + channels: 64 + layers: 3 + nheads: 8 + dropout: 0.1 + attr_dim: 64 + num_attr_heads: 4 + diffusion_steps: 50 + noise_schedule: quad + beta_start: 0.0001 + beta_end: 0.5 + base_patch: 4 + attr_proj_type: avg + attr_heads: 8 + attr_layers: 2 + n_attrs: 4 + feature_ratio: 0.5 + side_var_emb: 16 + side_time_emb: 112 +condition: + text: + enabled: false + input_dim: 1024 + embedding_key: cap_emb + dropout: 0.0 + cfg_scale: 1.0 + attribute: + enabled: true + continuous_dim: 0 + discrete_configs: + - num_classes: 4 + - num_classes: 2 + - num_classes: 4 + - num_classes: 4 + output_dim: 128 + dropout: 0.0 + label: + enabled: false + num_classes: 10 + output_dim: 64 + fusion: concat + condition_dropout: 0.0 +eval: + n_samples: 10 + metrics: + - dtw + - wape + - ed + - crps + - acd + - sd + - kd + - mdd + - fid + - sfd + - jftsd + - prdc_f1 + - joint_prdc_f1 + - cttp + - disc_auc + batch_size: 128 + save_samples: true + sampler: ddim + display_interval: 10 + clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-m/model_configs.yaml + clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-m/clip_model_best.pth + cache_folder: __CACHE_ROOT__/timeweaver/synth-m/seed2 + use_longalign: false + reference_split: train + disc_auc: + k_folds: 5 + epochs: 5 + batch_size: 128 + lr: 0.001 + weight_decay: 0.0001 + hidden_channels: 128 + num_layers: 3 + dropout: 0.1 + seed: 42 + num_workers: 0 + prdc: + k: 5 + max_samples: 5000 + seed: 0 + metric: euclidean + backend: auto + knn_chunk_size: 4096 + joint_enable: true + joint_weights: + ts: 1.0 + text: 1.0 + joint_normalize: standard + segment_classifier: + enable: false + segment_len: 128 + n_segments: 3 + seed: 42 + acd_max_lag: 50 + mdd_bins: 32 + use_cache: false + cache_file: predictions_cache.pkl + viz: + enable: true + k_cases: 20 + max_vars: 8 + ncols: 5 + seed: 42 + dpi: 200 + alpha: 0.25 + figsize_per_subplot: + - 3.0 + - 2.0 + output_type: pdf +output_dir: '' +created_at: '2025-12-27T20:29:42.287243' diff --git a/experiments/timeweaver/synth-m/seed2/results/expected_seed_metrics.json b/experiments/timeweaver/synth-m/seed2/results/expected_seed_metrics.json new file mode 100644 index 0000000000000000000000000000000000000000..94ce1b6329287fa3f58151a67f12f86a3d129a24 --- /dev/null +++ b/experiments/timeweaver/synth-m/seed2/results/expected_seed_metrics.json @@ -0,0 +1,21 @@ +{ + "dtw": 9.699557715230592, + "wape": 87.62276913723251, + "ed": 13.052642028808593, + "crps": 0.46461276566655224, + "acd": 0.0657203197479248, + "sd": 0.07425538727359915, + "kd": 0.10179796768961902, + "mdd": 0.016617431640625, + "fid": 34.594900900322614, + "sfd": 29.123449442252863, + "jftsd": 43.98960620155515, + "prdc_f1.f1": 0.38014822386915403, + "prdc_f1.precision": 0.72, + "prdc_f1.recall": 0.25825, + "joint_prdc_f1.f1": 0.44591782312925177, + "joint_prdc_f1.precision": 0.538, + "joint_prdc_f1.recall": 0.38075, + "cttp": 19.827544235229492, + "disc_auc": 0.991674375 +} \ No newline at end of file diff --git a/experiments/timeweaver/synth-m/seed2/summary.json b/experiments/timeweaver/synth-m/seed2/summary.json new file mode 100644 index 0000000000000000000000000000000000000000..2a3f636a8171fbab9c9c39efa7a265ae10435083 --- /dev/null +++ b/experiments/timeweaver/synth-m/seed2/summary.json @@ -0,0 +1,5 @@ +{ + "status": "completed", + "finished_at": "2025-12-27T21:06:35.535175", + "best_checkpoint": "checkpoints/finetune/best.ckpt" +} \ No newline at end of file diff --git a/experiments/ttscgan/synth-m/seed0/checkpoints/finetune/best.ckpt b/experiments/ttscgan/synth-m/seed0/checkpoints/finetune/best.ckpt new file mode 100644 index 0000000000000000000000000000000000000000..4f669300c4aa51423daf39976bbef1cab90fa0ce --- /dev/null +++ b/experiments/ttscgan/synth-m/seed0/checkpoints/finetune/best.ckpt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:10692451a3a5d9ee6fbca10819be6ba930ea7ec9c6a7e96e99f5164941d31ec5 +size 3776188 diff --git a/experiments/ttscgan/synth-m/seed1/checkpoints/finetune/best.ckpt b/experiments/ttscgan/synth-m/seed1/checkpoints/finetune/best.ckpt new file mode 100644 index 0000000000000000000000000000000000000000..6a6869028afd883bab77559c5cd17387752d34b4 --- /dev/null +++ b/experiments/ttscgan/synth-m/seed1/checkpoints/finetune/best.ckpt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3e5de4a967a4c2a680e6eafd3fea48702d45810d0505c791f4f9e5d55688cd1b +size 3776188 diff --git a/experiments/ttscgan/synth-m/seed2/checkpoints/finetune/best.ckpt b/experiments/ttscgan/synth-m/seed2/checkpoints/finetune/best.ckpt new file mode 100644 index 0000000000000000000000000000000000000000..4434ec0e35318a57eb64368e74717e2d6ad3d795 --- /dev/null +++ b/experiments/ttscgan/synth-m/seed2/checkpoints/finetune/best.ckpt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d906418ba22cfb3d39a914585197634b1b5763dc9b047de61d4e3ab2597c3187 +size 3776188 diff --git a/experiments/ttscgan/synth-u/seed0/checkpoints/finetune/best.ckpt b/experiments/ttscgan/synth-u/seed0/checkpoints/finetune/best.ckpt new file mode 100644 index 0000000000000000000000000000000000000000..dba5faa0c1078520788a329c8aa52e4304f20d8f --- /dev/null +++ b/experiments/ttscgan/synth-u/seed0/checkpoints/finetune/best.ckpt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0a1ae680dcc4b422146ad5d84e6f8d9801a91af1548fe621e57c7905583a64f2 +size 3701436 diff --git a/experiments/ttscgan/synth-u/seed1/checkpoints/finetune/best.ckpt b/experiments/ttscgan/synth-u/seed1/checkpoints/finetune/best.ckpt new file mode 100644 index 0000000000000000000000000000000000000000..be5b2d472c0c13359fb944b30457a94b145c5a7f --- /dev/null +++ b/experiments/ttscgan/synth-u/seed1/checkpoints/finetune/best.ckpt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:222d90e0469bc76569070e34cd20e3b56fa6c2a43f6f7dbce84c730596e2def6 +size 3701244 diff --git a/experiments/ttscgan/synth-u/seed2/checkpoints/finetune/best.ckpt b/experiments/ttscgan/synth-u/seed2/checkpoints/finetune/best.ckpt new file mode 100644 index 0000000000000000000000000000000000000000..28b29307dca92ec7930421c06e3c17eb3aab64c0 --- /dev/null +++ b/experiments/ttscgan/synth-u/seed2/checkpoints/finetune/best.ckpt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:79d712e78a5a65fee30dae9e13d4625f0188cb784b1794622730765e3609d68f +size 3701244 diff --git a/resources/cttp/synth-m/model_configs.yaml b/resources/cttp/synth-m/model_configs.yaml new file mode 100644 index 0000000000000000000000000000000000000000..19d64ca26bf90576a9d66104a7acc0d58d1fdd81 --- /dev/null +++ b/resources/cttp/synth-m/model_configs.yaml @@ -0,0 +1,29 @@ +clip_type: clip_patchtst +device: cuda:0 +loss_type: CE +text: + coemb_dim: 512 + device: cuda:0 + llm_finetune: frozen + output_type: cls + pretrain_model_dim: 768 + pretrain_model_path: ${LONGCLIP_ROOT} + textemb_hidden_dim: 1024 +ts: + activation: gelu + coemb_dim: 512 + d_ff: 256 + d_model: 64 + device: cuda:0 + dropout: 0.1 + e_layers: 2 + factor: 1 + n_heads: 8 + n_var: 2 + output_attention: true + padding: 0 + patch_len: 32 + pretrain_encoder_path: '' + seq_len: 128 + stride: 32 + type: patchtst_mae_pretrain diff --git a/resources/cttp/synth-u/model_configs.yaml b/resources/cttp/synth-u/model_configs.yaml new file mode 100644 index 0000000000000000000000000000000000000000..62d1499baa7c6e4337171767d472169efc304762 --- /dev/null +++ b/resources/cttp/synth-u/model_configs.yaml @@ -0,0 +1,28 @@ +clip_type: clip_patchtst +device: cuda:0 +loss_type: CE +text: + coemb_dim: 512 + device: cuda:0 + llm_finetune: frozen + pretrain_model_dim: 768 + pretrain_model_path: ${LONGCLIP_ROOT} + textemb_hidden_dim: 1024 +ts: + activation: gelu + coemb_dim: 512 + d_ff: 256 + d_model: 128 + device: cuda:0 + dropout: 0.1 + e_layers: 2 + factor: 1 + n_heads: 8 + n_var: 1 + output_attention: true + padding: 8 + patch_len: 16 + pretrain_encoder_path: '' + seq_len: 128 + stride: 8 + type: patchtst_mae_pretrain