| { | |
| "model": "sage", | |
| "feature": "bert", | |
| "in_dim": 768, | |
| "hidden": 128, | |
| "layers": 2, | |
| "heads": 4, | |
| "dropout": 0.5, | |
| "edge_drop": 0.2, | |
| "admm_p": 2.5, | |
| "admm_lambda": 1.0, | |
| "admm_iterations": 2, | |
| "feature_l1": 0.001, | |
| "epochs": 200, | |
| "batch_size": 128, | |
| "lr": 0.001, | |
| "weight_decay": 0.0001 | |
| } |