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)