I am currently training a model which is a mix of graph neural networks and LSTM. However that means for each of my training sample, I need to pass in a list of graphs. The current batch class in torch_geometric supports batching with torch_geometric.data.Batch.from_data_list() but this only allows one graph for each data point. How else can I go about batching the graphs when each data consist of multiple graphs?
Do you want to post the question/request to torch_geometric repo? We are not familiar with that.
ah I didn’t think of that, thanks for the suggestion!