examples/pytorch/graphsage/experimental/train_dist.py · 4889c5782290f1990c924fbea14ba904a3248231 · OpenDAS / dgl

Add 'nccl' backend in train_dist.py and fix pad_data function cuda bug (#3607) · 4889c578

xcwan authored Dec 24, 2021



* Add nccl backend  and fix pad_data function cuda bug

* Update train_dist.py

* Update train_dist.py
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>

4889c578

train_dist.py 15 KB

Replace train_dist.py