def build_vqae2_newatt(dataset, num_hid, att_dim, dec_dim):
w_emb = WordEmbedding(dataset.question_dictionary.ntoken, 300, 0.0)
q_emb = QuestionEmbedding(300, num_hid, 1, False, 0.0)
v_att = NewAttention(dataset.v_dim, q_emb.num_hid, num_hid)
q_net = FCNet([q_emb.num_hid, num_hid])
v_net = FCNet([dataset.v_dim, num_hid])
classifier = SimpleClassifier(
num_hid, num_hid * 2, dataset.num_ans_candidates, 0.5)
generator = STDecoder(
dataset.v_dim, num_hid, 300, dec_dim,\
dataset.explanation_dictionary.ntoken, 1, 0.5)
e_emb = ExplainEmbedding(generator.embed, 300, num_hid, 1, False, 0.0)
e_net = FCNet([e_emb.num_hid, num_hid])
return VQAE2(w_emb, q_emb, v_att, q_net, v_net, classifier, generator, e_emb, e_net)
def build_vqae3_split(dataset, num_hid, att_dim, dec_dim):
w_emb = WordEmbedding(dataset.question_dictionary.ntoken, 300, 0.0)
q_emb = QuestionEmbedding(300, num_hid, 1, False, 0.0)
v_att_1 = NewAttention(dataset.v_dim, q_emb.num_hid, num_hid)
q_net_1 = FCNet([q_emb.num_hid, num_hid])
v_net_1 = FCNet([dataset.v_dim, num_hid])
v_att_2 = NewAttention(dataset.v_dim, q_emb.num_hid, num_hid)
q_net_2 = FCNet([q_emb.num_hid, num_hid])
v_net_2 = FCNet([dataset.v_dim, num_hid])
classifier = SimpleClassifier(num_hid, num_hid * 2,
dataset.num_ans_candidates, 0.5)
generator = STDecoder(
dataset.v_dim, num_hid, 300, dec_dim,\
dataset.explanation_dictionary.ntoken, 1, 0.5)
e_emb = ExplainEmbedding(generator.embed, 300, num_hid, 1, False, 0.0,
'GRU')
T_vq = FCNet([num_hid, num_hid])
T_e = FCNet([e_emb.num_hid, num_hid])
return Split_VQAE(w_emb, q_emb, v_att_1, q_net_1, v_net_1, v_att_2,
q_net_2, v_net_2, classifier, generator, e_emb, T_vq,
T_e)