debugging

mlcommons · Jan 3, 2024 · d6716df · d6716df
1 parent 503e4f0
commit d6716df
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/algorithmic_efficiency/workloads/ogbg/ogbg_pytorch/models.py b/algorithmic_efficiency/workloads/ogbg/ogbg_pytorch/models.py
@@ -181,6 +181,8 @@ def forward(self, graph: GraphsTuple) -> GraphsTuple:
           lambda g: torch.repeat_interleave(g, n_node, dim=0), globals_)
       print('SHAPES')
       print(nodes.shape, sent_attributes.shape, received_attributes.shape, global_attributes.shape)
+      print(senders.shape)
+      print(receivers.shape)
       node_fn_inputs = torch.cat(
           [nodes, sent_attributes, received_attributes, global_attributes],
           dim=-1)