GANA-FewShotKGC
GANA-FewShotKGC copied to clipboard
想问一下,neibor_encoder 最后一步相加时 两个参数尺寸不对是因为什么 或者怎么解决
gate_tem = self.gate_w(out_attn) + self.gate_b
gate = torch.sigmoid(gate_tem)
out_neighbor = torch.mul(out_attn, gate)
out_neighbor = out_neighbor + entself_embeds 这里一个是 bs*100 一个是bs*50*100
是因为我中间的参数没有设置,还是因为什么问题呢 请大佬们在不忙的时候,帮忙解决一下,谢谢
请查看#7,下载最新的model_gana.py和数据集,就可以运行代码。