Skip to content

GitLab

Unverified Commit ed9ef14a authored by Ramon Zhou's avatar Ramon Zhou Committed by GitHub
Browse files

[Misc] Turn off `fused` in DGL rgcn example (#6977)

parent cf8fea73
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
examples/core/rgcn/hetero_rgcn.py
View file @ ed9ef14a
...@@ -116,7 +116,7 @@ def prepare_data(args, device): ...@@ -116,7 +116,7 @@ def prepare_data(args, device):
# Initialize a train sampler that samples neighbors for multi-layer graph # Initialize a train sampler that samples neighbors for multi-layer graph
# convolution. It samples 25 and 10 neighbors for the first and second # convolution. It samples 25 and 10 neighbors for the first and second
# layers respectively. # layers respectively.
sampler = dgl.dataloading.MultiLayerNeighborSampler([25, 10]) sampler = dgl.dataloading.MultiLayerNeighborSampler([25, 10], fused=False)
num_workers = args.num_workers num_workers = args.num_workers
train_loader = dgl.dataloading.DataLoader( train_loader = dgl.dataloading.DataLoader(
g, g,
...@@ -488,7 +488,7 @@ def evaluate( ...@@ -488,7 +488,7 @@ def evaluate(
else: else:
evaluator = MAG240MEvaluator() evaluator = MAG240MEvaluator()
sampler = dgl.dataloading.MultiLayerNeighborSampler([25, 10]) sampler = dgl.dataloading.MultiLayerNeighborSampler([25, 10], fused=False)
dataloader = dgl.dataloading.DataLoader( dataloader = dgl.dataloading.DataLoader(
g, g,
idx, idx,
......
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