Skip to content

GitLab

Unverified Commit a6abffe3 authored by nv-dlasalle's avatar nv-dlasalle Committed by GitHub
Browse files

[Fix] Don't create extra formats in dataloader (#2551) 


Co-authored-by: default avatarMinjie Wang <wmjlyjemaine@gmail.com>
parent 2288098b
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 2 deletions
python/dgl/dataloading/dataloader.py
View file @ a6abffe3
...@@ -251,8 +251,6 @@ class BlockSampler(object): ...@@ -251,8 +251,6 @@ class BlockSampler(object):
seed_nodes = {ntype: block.srcnodes[ntype].data[NID] for ntype in block.srctypes} seed_nodes = {ntype: block.srcnodes[ntype].data[NID] for ntype in block.srctypes}
# Pre-generate CSR format so that it can be used in training directly
block.create_formats_()
blocks.insert(0, block) blocks.insert(0, block)
return blocks return blocks
......
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