model.pth_config.json

{
  "num_nodes": 207,
  "gcn_depth": 2,
  "dropout": 0.3,
  "subgraph_size": 20,
  "node_dim": 40,
  "dilation_exponential": 1,
  "conv_channels": 32,
  "residual_channels": 32,
  "skip_channels": 64,
  "end_channels": 128,
  "in_dim": 2,
  "seq_in_len": 12,
  "seq_out_len": 12,
  "layers": 3,
  "propalpha": 0.05,
  "tanhalpha": 3.0,
  "batch_size": 64,
  "val_batch_size": 8,
  "epochs": 100,
  "learning_rate": 0.001,
  "weight_decay": 0.0001,
  "clip": 5,
  "step_size1": 2500,
  "step_size2": 100,
  "gcn_true": true,
  "buildA_true": true,
  "cl": true,
  "device": "cuda",
  "seed": 101,
  "print_every": 999999,
  "save": "/workspace/traffic-forecasting/models/MTGNN/model_metr-la/"
}