ENZYMES = {
    "experiment.batch_size": 256,
    "experiment.normalization": True,
    "optimizer.lr": 0.003,
    "optimizer.weight_decay": 0.0,
    "model.hidden_channels": 64,
    "model.num_layers": 3,
    "model.num_lin_layers": 1,
    "model.num_rnn_layers": 0,
    "model.dropout": 0.0,
    "model.glob_pool_mode": "mean",
    "model.act": "tanh",
}

PROTEINS = {
    "experiment.batch_size": 512,
    "experiment.normalization": True,
    "optimizer.lr": 0.003,
    "optimizer.weight_decay": 0.0,
    "model.hidden_channels": 32,
    "model.num_layers": 3,
    "model.num_lin_layers": 3,
    "model.num_rnn_layers": 0,
    "model.dropout": 0.0,
    "model.glob_pool_mode": "mean",
    "model.act": "tanh",
}

CORA = {
    "experiment.batch_size": 512,
    "experiment.normalization": True,
    "optimizer.lr": 0.001,
    "optimizer.weight_decay": 0.0,
    "model.hidden_channels": 64,
    "model.num_layers": 2,
    "model.num_lin_layers": 1,
    "model.num_rnn_layers": 0,
    "model.dropout": 0.0,
    "model.glob_pool_mode": None,
    "model.act": "tanh",
}
