examples/pytorch/graphsage/experimental/train_dist_unsupervised.py · 33a8bb93bf4c6d6adc2dcb252c07ee3572616c49 · 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