fix acc drop bug

Author: BowenYao18

examples/graphbolt/rgcn/hetero_rgcn.py

@@ -163,7 +163,7 @@ def extract_embed(node_embed, input_nodes):
 
 def extract_node_features(name, block, data, node_embed, device):
     """Extract the node features from embedding layer or raw features."""
-    if name == "ogbn-mag" or "igb-het" in name:
+    if name == "ogbn-mag":
         input_nodes = {
             k: v.to(device) for k, v in block.srcdata[dgl.NID].items()
         }
@@ -288,13 +288,6 @@ def __init__(
             }
         )
 
-        self.loop_weights = nn.ModuleDict(
-            {
-                ntype: nn.Linear(in_size, out_size, bias=True)
-                for ntype in self.ntypes
-            }
-        )
-
         self.dropout = nn.Dropout(dropout)
         # Initialize parameters of the model.
         self.reset_parameters()
@@ -677,6 +670,8 @@ def main(args):
             "igb-het-small",
             "igb-het-medium",
             "igb-het-large",
+            "igb-het",
+            "igb-het-MLPerf"
         ],
         help="Dataset name. Possible values: ogbn-mag, ogb-lsc-mag240m, "
         " igb-het-[tiny|small|medium|large].",