add somecomments

2024-06-08 21:35:35 +02:00
parent 2c00828630
commit 4f8945ca07
6 changed files with 72 additions and 3 deletions
--- a/graph_dit/datasets/abstract_dataset.py
+++ b/graph_dit/datasets/abstract_dataset.py
@@ -123,4 +123,8 @@ class AbstractDatasetInfos:
                           'y': example_batch['y'].size(1)}
        self.output_dims = {'X': example_batch_x.size(1),
                            'E': example_batch_edge_attr.size(1),
-                            'y': example_batch['y'].size(1)}
+                            'y': example_batch['y'].size(1)}
+        print('input dims')
+        print(self.input_dims)
+        print('output dims')
+        print(self.output_dims)