Source code for gigl.src.common.constants.training

from torch.nn import functional as F


[docs]
ACT_MAP = {
    "relu": F.relu,
    "elu": F.elu,
    "leakyrelu": F.leaky_relu,
    "sigmoid": F.sigmoid,
    "tanh": F.tanh,
    "gelu": F.gelu,
    "silu": F.silu,
}


[docs]
DEFAULT_NUM_GNN_HOPS = 2