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):
tqdm.tqdm(enumerate(dataloader)) if rank == 0 else enumerate(dataloader)
):
blocks = data.blocks
x = data.node_features["feat"].float()
x = data.node_features["feat"]
y.append(data.labels)
y_hats.append(model.module(blocks, x))
......@@ -228,7 +228,7 @@ def train(
):
# The input features are from the source nodes in the first
# layer's computation graph.
x = data.node_features["feat"].float()
x = data.node_features["feat"]
# The ground truth labels are from the destination nodes
# in the last layer's computation graph.
......
......@@ -529,6 +529,7 @@ class BuiltinDataset(OnDiskDataset):
.. note::
Reverse edges are added to the original graph.
Node features are stored as float32.
**ogb-lsc-mag240m**
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