{
  "num_nodes": 207,
  "in_steps": 12,
  "out_steps": 12,
  "input_dim": 2,
  "output_dim": 1,
  "steps_per_day": 288,
  "input_embedding_dim": 24,
  "tod_embedding_dim": 24,
  "dow_embedding_dim": 0,
  "adaptive_embedding_dim": 80,
  "num_heads": 4,
  "num_layers": 3,
  "dropout": 0.1,
  "dropout_a": 0.3,
  "kernel_size": [
    1
  ],
  "epochs": 100,
  "batch_size": 64,
  "learning_rate": 0.001,
  "weight_decay": 0.0003,
  "milestones": [
    20,
    30
  ],
  "lr_decay_rate": 0.1,
  "early_stop": 10,
  "clip_grad": 0,
  "device": "cuda",
  "verbose": 1
}