basic:
  dataset: "PEMSD7"
  mode: "train"
  device: "cuda:0"
  model: "STGNCDE"

data:
  num_nodes: 883
  lag: 12
  horizon: 12
  val_ratio: 0.2
  test_ratio: 0.2
  tod: false
  normalizer: std
  column_wise: false
  default_graph: true
  steps_per_day: 288
  days_per_week: 7

model:
  type: type1
  g_type: agc
  input_dim: 2
  output_dim: 1
  embed_dim: 10
  hid_dim: 64
  hid_hid_dim: 64
  num_layers: 2
  cheb_k: 2
  solver: rk4

train:
  loss_func: mae
  seed: 10
  batch_size: 64
  epochs: 100
  lr_init: 0.003
  lr_decay: false
  weight_decay: 0
  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

test:
  mae_thresh: null
  mape_thresh: 0.0

log:
  log_step: 200
  plot: false