SBMs_node_clustering_GIN_CLUSTER_500k.json

{
    "gpu": {
        "use": true,
        "id": 0
    },
    
    "model": "GIN",
    "dataset": "SBM_CLUSTER",
    
    "out_dir": "out/SBMs_node_classification/",
    
    "params": {
        "seed": 41,
        "epochs": 1000,
        "batch_size": 128,
        "init_lr": 0.001,
        "lr_reduce_factor": 0.5,
        "lr_schedule_patience": 5,
        "min_lr": 1e-5,
        "weight_decay": 0.0,
        "print_epoch_interval": 5,
        "max_time": 12
    },
    
    "net_params": {
        "L": 16,
        "hidden_dim": 124,
        "residual": true,
        "readout": "sum",
        "n_mlp_GIN": 2,
        "learn_eps_GIN": true,
        "neighbor_aggr_GIN": "sum",
        "in_feat_dropout": 0.0,
        "dropout": 0.0,
        "batch_norm": true
    }
}