basic:
  dataset: METR-LA
  device: cuda:0
  mode: train
  model: STID
  seed: 2023

data:
  batch_size: 64
  column_wise: false
  days_per_week: 7
  horizon: 24
  input_dim: 1
  lag: 24
  normalizer: std
  num_nodes: 207
  steps_per_day: 288
  test_ratio: 0.2
  val_ratio: 0.2

model:
  batch_size: 64
  day_of_week_size: 7
  embed_dim: 32
  history: 24
  horizon: 24
  if_D_i_W: true
  if_T_i_D: true
  if_node: true
  input_dim: 3
  input_len: 24
  node_dim: 32
  num_layer: 3
  num_nodes: 207
  output_dim: 1
  output_len: 24
  temp_dim_diw: 32
  temp_dim_tid: 32
  time_of_day_size: 288

train:
  batch_size: 64
  debug: true
  early_stop: true
  early_stop_patience: 15
  epochs: 300
  grad_norm: false
  log_step: 200
  loss_func: mae
  lr_decay: false
  lr_decay_rate: 0.3
  lr_decay_step: 1,50,80
  lr_init: 0.002
  mae_thresh:
  mape_thresh: 0.0
  max_grad_norm: 5
  output_dim: 1
  plot: false
  real_value: true
  seed: 1
  weight_decay: 0.0001