From 44a17f37eced148bc95b553390b70ce35d14bbbb Mon Sep 17 00:00:00 2001 From: riship Date: Sat, 11 Jan 2025 10:08:42 -0800 Subject: [PATCH] debug --- torch_geometric/data/large_graph_indexer.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/torch_geometric/data/large_graph_indexer.py b/torch_geometric/data/large_graph_indexer.py index 49abbbcf306f..849dbe6ff92b 100644 --- a/torch_geometric/data/large_graph_indexer.py +++ b/torch_geometric/data/large_graph_indexer.py @@ -539,9 +539,12 @@ def to_data(self, node_feature_name: str, node_id = torch.LongTensor(range(len(x))) print("self.get_edge_features(EDGE_INDEX)[:10]=", self.get_edge_features(EDGE_INDEX)[:10]) + print("self.get_edge_features(EDGE_INDEX)[-10:]=", + self.get_edge_features(EDGE_INDEX)[-10:]) edge_index = torch.t( torch.LongTensor(self.get_edge_features(EDGE_INDEX))) - print("edge_index tensor of full graph=", edge_index[:, :10]) + print("edge_index tensor of full graph[:10]=", edge_index[:, :10]) + print("edge_index tensor of full graph[-10:]=", edge_index[:, -10:]) edge_attr = (self.get_edge_features(edge_feature_name) if edge_feature_name is not None else None)