"git@developer.sourcefind.cn:change/sglang.git" did not exist on "564bdf29f7ef630f4ff8cd32afa5c070e99e67f0"
Unverified Commit 325e67ce authored by Ramon Zhou's avatar Ramon Zhou Committed by GitHub
Browse files

[Misc] Change node features type of ogbn-products to float32 (#6518)

parent 50c5e488
...@@ -164,7 +164,7 @@ def evaluate(rank, args, model, graph, features, itemset, num_classes, device): ...@@ -164,7 +164,7 @@ def evaluate(rank, args, model, graph, features, itemset, num_classes, device):
tqdm.tqdm(enumerate(dataloader)) if rank == 0 else enumerate(dataloader) tqdm.tqdm(enumerate(dataloader)) if rank == 0 else enumerate(dataloader)
): ):
blocks = data.blocks blocks = data.blocks
x = data.node_features["feat"].float() x = data.node_features["feat"]
y.append(data.labels) y.append(data.labels)
y_hats.append(model.module(blocks, x)) y_hats.append(model.module(blocks, x))
...@@ -228,7 +228,7 @@ def train( ...@@ -228,7 +228,7 @@ def train(
): ):
# The input features are from the source nodes in the first # The input features are from the source nodes in the first
# layer's computation graph. # layer's computation graph.
x = data.node_features["feat"].float() x = data.node_features["feat"]
# The ground truth labels are from the destination nodes # The ground truth labels are from the destination nodes
# in the last layer's computation graph. # in the last layer's computation graph.
......
...@@ -529,6 +529,7 @@ class BuiltinDataset(OnDiskDataset): ...@@ -529,6 +529,7 @@ class BuiltinDataset(OnDiskDataset):
.. note:: .. note::
Reverse edges are added to the original graph. Reverse edges are added to the original graph.
Node features are stored as float32.
**ogb-lsc-mag240m** **ogb-lsc-mag240m**
The ogb-lsc-mag240m dataset is a heterogeneous academic graph extracted The ogb-lsc-mag240m dataset is a heterogeneous academic graph extracted
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment