From b9dc9d1c3beae1969f76188e22d09610177e5ee4 Mon Sep 17 00:00:00 2001 From: song Date: Sat, 29 Jul 2023 03:14:57 +0000 Subject: [PATCH] modify PaddlePaddle GCN result Signed-off-by: song --- gnn/GCN/paddlepaddle/README.md | 18 ++++++++++++------ 1 file changed, 12 insertions(+), 6 deletions(-) diff --git a/gnn/GCN/paddlepaddle/README.md b/gnn/GCN/paddlepaddle/README.md index b44346cc..c713af33 100644 --- a/gnn/GCN/paddlepaddle/README.md +++ b/gnn/GCN/paddlepaddle/README.md @@ -15,9 +15,9 @@ git clone https://github.com/PaddlePaddle/PGL.git ``` # Pip the requirements -pip install pgl -pip install urllib3==1.23 -pip install networkx +pip3 install pgl +pip3 install urllib3==1.23 +pip3 install networkx ``` ## Step 2:Preparing datasets @@ -35,16 +35,22 @@ cd PGL/examples/gcn/ CUDA_VISIBLE_DEVICES=0 python3 train.py --dataset cora # Run on CPU -CUDA_VISIBLE_DEVICES= python3 train.py --dataset cora +python3 train.py --dataset cora ``` ## Results +| GPUS | Datasets | speed | +| ---------- | -------- | -------- | +| BI V100×1 | CORA | 0.0064 | +| BI V100×1 | PUBMED | 0.0076 | +| BI V100×1 | CITESEER | 0.0085 | + | GPUS | Datasets | Accurary | | ---------- | -------- | -------- | | BI V100×1 | CORA | 80.3% | -| BI V100×1 | PUBMED | 79.3% | -| BI V100×1 | CITESEER | 71.0% | +| BI V100×1 | PUBMED | 79.0% | +| BI V100×1 | CITESEER | 70.6% | ## Reference -- Gitee