basic:
  device: cuda:0
  dataset: PEMS08
  model: STDEN
  mode: train
  seed: 2025

data:
  dataset_dir: data/PEMS08
  val_batch_size: 32
  graph_pkl_filename: data/PEMS08/PEMS08_spatial_distance.npy
  num_nodes: 170
  batch_size: 32
  input_dim: 1
  lag: 24
  horizon: 24
  val_ratio: 0.2
  test_ratio: 0.2
  tod: False
  normalizer: std
  column_wise: False
  default_graph: True
  add_time_in_day: True
  add_day_in_week: True
  steps_per_day: 24
  days_per_week: 7

model:
  l1_decay: 0
  seq_len: 12
  horizon: 12
  input_dim: 1 
  output_dim: 1
  latent_dim: 4
  n_traj_samples: 3
  ode_method: dopri5 
  odeint_atol: 0.00001 
  odeint_rtol: 0.00001 
  rnn_units: 64 
  num_rnn_layers: 1
  gcn_step: 2
  filter_type: default # unkP IncP default
  recg_type: gru 
  save_latent: false
  nfe: false

train:
  loss: mae
  batch_size: 64
  epochs: 100
  lr_init: 0.003
  mape_thresh: 0.001
  mae_thresh: None
  debug: False
  output_dim: 1
  weight_decay: 0
  lr_decay: False
  lr_decay_rate: 0.3
  lr_decay_step: "5,20,40,70"
  early_stop: True
  early_stop_patience: 15
  grad_norm: False
  max_grad_norm: 5
  real_value: True