examples/pytorch/graphsage/train_sampling_multi_gpu.py · 9497a9be85cf03e2a21e28159b59f51cba9d4aad · OpenDAS / dgl

[Bugfix][Examples] Fix graphsage multigpu training example training set size (#3002) · 9497a9be

nv-dlasalle authored Jun 10, 2021



* Make multigpu graphsage use whole datset

* Specify queeze dimension

* Remove squeeze dimension
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>

9497a9be

train_sampling_multi_gpu.py 8.86 KB

Replace train_sampling_multi_gpu.py