Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
dgl
Commits
325e67ce
Unverified
Commit
325e67ce
authored
Nov 02, 2023
by
Ramon Zhou
Committed by
GitHub
Nov 02, 2023
Browse files
[Misc] Change node features type of ogbn-products to float32 (#6518)
parent
50c5e488
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
3 additions
and
2 deletions
+3
-2
examples/multigpu/graphbolt/node_classification.py
examples/multigpu/graphbolt/node_classification.py
+2
-2
python/dgl/graphbolt/impl/ondisk_dataset.py
python/dgl/graphbolt/impl/ondisk_dataset.py
+1
-0
No files found.
examples/multigpu/graphbolt/node_classification.py
View file @
325e67ce
...
@@ -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.
...
...
python/dgl/graphbolt/impl/ondisk_dataset.py
View file @
325e67ce
...
@@ -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
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment