update node/edge encoder naming

JiaxuanYou · JiaxuanYou · commit 77f1e7acb4d0 · 2020-11-24T16:03:32.000-08:00
diff --git a/graphgym/models/layer.py b/graphgym/models/layer.py
@@ -102,7 +102,7 @@ def __init__(self, dim_in):
         self.bn = nn.BatchNorm1d(dim_in, eps=cfg.bn.eps, momentum=cfg.bn.mom)
 
     def forward(self, batch):
-        batch.node_feature = self.bn(batch.edge_feature)
+        batch.edge_feature = self.bn(batch.edge_feature)
         return batch
 
 
