examples/pytorch/graphsage/link_pred.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

link_pred.py 7.08 KB

Replace link_pred.py