运行问题
Closed this issue · 9 comments
850587600 commented
运行代码的时候总是走到第二轮就内存爆了,想知道是什么原因
lzxlin commented
资源不足可以考虑适当减少batch size或者通过梯度累积实现较大的batch size
850587600 commented
我把bsize改成1都还是不行,大佬方便给个联系方式吗0.0
lzxlin commented
你使用的内存或者显存是多少呢?可以检查下是否是GAT占有资源过多的问题,或者把GAT换成GCN试下
850587600 commented
16G运行内存,我是在服务器上跑的我不知道怎么在服务器上看GAT是否占有资源过多
lzxlin commented
可以试着调小hidden_units参数,降低GAT的隐层大小,或者把structure_encoder改成"gcn"
850587600 commented
好滴我试下谢谢大佬,不过改成了GCN之后效果是不是就不太好了
lzxlin commented
实验效果GAT会比GCN好些,但是GCN速度会更快
850587600 commented
大佬我想知道关于属性和关系、实体名嵌入看你的文章好像都是直接送往简单的前馈层,得到嵌入,但是我看其他文章都是利用到了TransE获得关系嵌入和属性特征利用Bert获得嵌入,我想知道您这篇文章为什么这么设计。
lzxlin commented
特征的编码器不是本文的核心创新,所以采样了简单MLP层,可以灵活替换不同的编码器至你的项目中