Skip to content

GitLab

"src/git@developer.sourcefind.cn:OpenDAS/dgl.git" did not exist on "c3a33407fe8b4938156427c701f0386dfc1a38de"
Unverified Commit 0b902d03 authored by Quan (Andy) Gan's avatar Quan (Andy) Gan Committed by GitHub
Browse files

[Bug][Model] Bugfix in PinSAGE code (#1533)

parent 7b6af87a
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
examples/pytorch/pinsage/sampler.py
View file @ 0b902d03
...@@ -6,7 +6,9 @@ from torch.utils.data import IterableDataset, DataLoader ...@@ -6,7 +6,9 @@ from torch.utils.data import IterableDataset, DataLoader
def compact_and_copy(frontier, seeds): def compact_and_copy(frontier, seeds):
block = dgl.to_block(frontier, seeds) block = dgl.to_block(frontier, seeds)
for col, data in frontier.edata.items(): for col, data in frontier.edata.items():
block.edata[col] = data if col == dgl.EID:
continue
block.edata[col] = data[block.edata[dgl.EID]]
return block return block
class ItemToItemBatchSampler(IterableDataset): class ItemToItemBatchSampler(IterableDataset):
......
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