set batch_y to 1 and want to test 15625

2024-08-20 22:15:25 +02:00
parent 1fa2d49c11
commit 3950a8438d
2 changed files with 3 additions and 2 deletions
--- a/graph_dit/datasets/dataset.py
+++ b/graph_dit/datasets/dataset.py
@@ -781,7 +781,7 @@ class Dataset(InMemoryDataset):
                print(f'idx={idx}, y={y}')
                y = torch.tensor([0, 0], dtype=torch.float).view(1, -1)
                data = Data(x=x, edge_index=edge_index, edge_attr=edge_attr, y=y, idx=i)
-                return None
+                # return None
            return data
        graph_list = []
        class Args: