examples/pytorch/graphsage/experimental/train_dist_unsupervised.py · 283b2cfb96728376beee3df7763620f85b84b5c8 · OpenDAS / dgl

[Distributed] Fix issues of GraphSage to support running on GPU (#2084) · 283b2cfb

Chao Ma authored Aug 25, 2020



* fix issues on GPU

* update

* update

* update

* update

* update

* update

* update

* update

* update
Co-authored-by: Ma <manchao@38f9d3587685.ant.amazon.com>

283b2cfb

train_dist_unsupervised.py 20.4 KB

Replace train_dist_unsupervised.py