| { | |
| "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/", | |
| "dataset_name": "METR-LA" | |
| } |