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'