examples/pytorch/graphsage/node_classification.py · f931c6ba548537d112dbee0a709b6e0bf1fdbe64 · OpenDAS / dgl

[Examples] Add pure gpu mode in the GraphSAGE node classification and link prediction (#3856) · f931c6ba

Serge Panev authored Apr 11, 2022


Signed-off-by: Serge Panev <spanev@nvidia.com>
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>

f931c6ba

node_classification.py 4.88 KB

Replace node_classification.py