Open chang-l opened 2 years ago
Fixed by #4082
@jermainewang
In fact, this issue has not been resolved because of the outdated DGL dataloader APIs. It seems train_sampling.py
uses at least two features that have been deprecated:
(1) Set return_eids=True
in MultiLayerNeighborSampler
to use dgl.EID
later (no longer support in 0.8x)
(2) Specify g_sampling
in dataloader constructor (no longer support in 0.8x).
This issue has been automatically marked as stale due to lack of activity. It will be closed if no further activity occurs. Thank you
🐛 Bug
Mini-batch sampling pipline update, one of the major features released in v0.8, has not been propagated to examples/pytorch/gcmc/train_sampling.py.
To Reproduce
Step(s) to reproduce the behavior:
(single GPU run)
Traceback (most recent call last): File "/workspace/examples/gcmc/train_sampling.py", line 377, in
run(0, n_gpus, args, devices, dataset)
File "/workspace/examples/gcmc/train_sampling.py", line 202, in run
sampler = dgl.dataloading.MultiLayerNeighborSampler([None], return_eids=True)
TypeError: init() got an unexpected keyword argument 'return_eids'
Expected behavior
The case should run through (with single/multi-GPU settings)
Environment
conda
,pip
, source): sourceAdditional context
The same type of error (mini-batch sampling update) is also happened to all TGN examples (examples/pytorch/tgn/train.py), where EdgeCollator is no longer a valid class name from latest dgl-v0.8x release...