我目前正在训练一个混合了图神经网络和 LSTM 的模型。然而，这意味着对于我的每个训练样本，我需要传入一个图表列表。torch_geometric 中的当前批处理类支持批处理，torch_geometric.data.Batch.from_data_list()但这只允许每个数据点使用一个图表。我还能如何对图表进行批处理？
问问题
79 次
1 回答
0
使用对角批处理：
https://pytorch-geometric.readthedocs.io/en/latest/notes/batching.html
简单地说，您会将所有图作为子图放入一个大图中。所有子图都将被隔离。
请参阅 TUDataset 中的示例： https ://colab.research.google.com/drive/1I8a0DfQ3fI7Njc62__mVXUlcAleUclnb?usp=sharing
于 2022-02-03T21:06:44.237 回答