Spaces:

eaglelandsonce
/

TensorFlowClass

Sleeping

eaglelandsonce commited on Jul 7, 2024

Commit

eaa1227

verified ·

1 Parent(s): a786ddb

Update pages/15_Graphs.py

Files changed (1) hide show

pages/15_Graphs.py CHANGED Viewed

@@ -23,7 +23,7 @@ def model_fn(graph_tensor_spec: tfgnn.GraphTensorSpec):
         graph = mt_albis.MtAlbisGraphUpdate(
             units=128,
             message_dim=64,
-            receiver_tag=tfgnn.NODES,  # Correctly use tfgnn.NODES
             attention_type="none",
             simple_conv_reduce_type="mean",
             normalization_type="layer",

         graph = mt_albis.MtAlbisGraphUpdate(
             units=128,
             message_dim=64,
+            receiver_tag=tfgnn.NODES,  # Ensure it is correctly set to tfgnn.NODES
             attention_type="none",
             simple_conv_reduce_type="mean",
             normalization_type="layer",