def __init__(self, CONFIG, multi_attn=False):
super(H_O_EdgeApplyModule, self).__init__()
self.multi_attn = multi_attn
self.edge_fc = MLP(CONFIG.G_E_L_S, CONFIG.G_E_A, CONFIG.G_E_B,
CONFIG.G_E_BN, CONFIG.G_E_D)
self.edge_fc_lang = MLP(CONFIG.G_E_L_S2, CONFIG.G_E_A2, CONFIG.G_E_B2,
CONFIG.G_E_BN2, CONFIG.G_E_D2)
def __init__(self, CONFIG, HICO=None):
super(Predictor, self).__init__()
if not HICO:
self.classifier = MLP(CONFIG.G_ER_L_S, CONFIG.G_ER_A,
CONFIG.G_ER_B, CONFIG.G_ER_BN, CONFIG.G_ER_D)
else:
self.classifier = MLP(CONFIG.G_ER_L_S_HICO, CONFIG.G_ER_A_HICO,
CONFIG.G_ER_B_HICO, CONFIG.G_ER_BN_HICO,
CONFIG.G_ER_D_HICO)
self.sigmoid = nn.Sigmoid()
def __init__(self,
layer_sizes,
activation,
bias=True,
use_bn=True,
drop_prob=None):
super(TowMLPHead, self).__init__()
self.head_fc = MLP(layer_sizes, activation, bias, use_bn, drop_prob)
def __init__(self, CONFIG):
super(Predictor, self).__init__()
self.classifier = MLP(CONFIG.G_ER_L_S, CONFIG.G_ER_A, CONFIG.G_ER_B,
CONFIG.G_ER_BN, CONFIG.G_ER_D)
self.sigmoid = nn.Sigmoid()
def __init__(self, CONFIG):
super(NodeUpdate, self).__init__()
self.fc = MLP(CONFIG.G_N_L_S_U, CONFIG.G_N_A_U, CONFIG.G_N_B_U,
CONFIG.G_N_BN_U, CONFIG.G_N_D_U)
self.fc_lang = MLP(CONFIG.G_N_L_S2_U, CONFIG.G_N_A2_U, CONFIG.G_N_B2_U,
CONFIG.G_N_BN2_U, CONFIG.G_N_D2_U)
def __init__(self, CONFIG):
super(O_NodeApplyModule, self).__init__()
self.node_fc = MLP(CONFIG.G_N_L_S, CONFIG.G_N_A, CONFIG.G_N_B,
CONFIG.G_N_BN, CONFIG.G_N_D)
self.node_fc_lang = MLP(CONFIG.G_N_L_S2, CONFIG.G_N_A2, CONFIG.G_N_B2,
CONFIG.G_N_BN2, CONFIG.G_N_D2)
def __init__(self, CONFIG):
super(E_AttentionModule2, self).__init__()
self.attn_fc2 = MLP(CONFIG.G_A_L_S2, CONFIG.G_A_A2, CONFIG.G_A_B2,
CONFIG.G_A_BN2, CONFIG.G_A_D2)
def __init__(self, CONFIG):
super(E_AttentionModule1, self).__init__()
self.attn_fc = MLP(CONFIG.G_A_L_S, CONFIG.G_A_A, CONFIG.G_A_B,
CONFIG.G_A_BN, CONFIG.G_A_D)
self.attn_fc_lang = MLP(CONFIG.G_A_L_S2, CONFIG.G_A_A2, CONFIG.G_A_B2,
CONFIG.G_A_BN2, CONFIG.G_A_D2)