GAT 用于 图像分割

[tanjia123456]

您好，我就直接用中文了

我看您的代码也没设置batch-size，我跑他们的代码也是这样，似乎很容易造成内存不足。

另外，adj[np.newaxis]为什么要加这一个维度呢？我的理解哈，就相当于说明只有一个graph。我使用GAT来做图像分割，相当于有79个graph(subject)，每个graph有10242个节点，8个特征，分成76个类别。
在此我想问问我将adj,features,label 建成 3D (79，10242，10242) (79,10242,8) (79,10242,76),再根据graph的个数来划分训练集测试集，如果train中有52，test有27，则y_train形式（79，10242，76）79个中只有52个（10242，10242）,其余位置是0。 那train_mask就是(1,79)么，类型为布尔？我不太清楚train_mask这种形式正确不？

[taishan1994]

因为计算注意力的时候是两两节点之间计算的，因此不能使用Batchsize，当然你可以每次选择一些节点以及他们的邻居节点，这样就可以分批了，但是邻居节点的个数必须是一致的，也就是要设定一个值，你可以去看一下graph-bert，这篇论文可以进行Batchsize训练，邻居是根据重要程度选择。新增加一个维度因为输入要是[batch, features，nodes]这种格式的，1相当于占位，如果是分批的话这里就好理解了。y_train形式（79，10242，76），选择其中的52个，则mask应该是利用mask将这些进行提取出来，torch.index_select(outputs, 0, tr，train_mask)就是实现这个功能，train_mask中需要提取的位置就是True。

[tanjia123456]

感谢你的回复呀！我发现tensorflow版本的有batch-size，但是pytorch我还没找到开源的有batch-size的代码。------------------&nbsp;原始邮件&nbsp;------------------ 发件人:&nbsp;&quot;taishan1994&quot;<notifications@github.com&gt; 发送时间:&nbsp;2020年11月6日(星期五) 晚上9:35 收件人:&nbsp;&quot;taishan1994/pytorch_gat&quot;<pytorch_gat@noreply.github.com&gt;; 抄送:&nbsp;&quot;tanjia123456&quot;<1248742467@qq.com&gt;;&quot;Author&quot;<author@noreply.github.com&gt;; 主题:&nbsp;Re: [taishan1994/pytorch_gat] GAT 用于 图像分割 (#1)