From 69ccb4c43b768d30ad18f944b2973e48d0efd6b0 Mon Sep 17 00:00:00 2001 From: Zhong Hui Date: Fri, 18 Dec 2020 23:02:37 +0800 Subject: [PATCH] fix for pgl for the erniesage fix for pgl for the erniesage --- .../text_graph/erniesage/config/erniesage_link_prediction.yaml | 2 +- PaddleNLP/examples/text_graph/erniesage/data/dataset.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/PaddleNLP/examples/text_graph/erniesage/config/erniesage_link_prediction.yaml b/PaddleNLP/examples/text_graph/erniesage/config/erniesage_link_prediction.yaml index bde00ad2..3beb0691 100644 --- a/PaddleNLP/examples/text_graph/erniesage/config/erniesage_link_prediction.yaml +++ b/PaddleNLP/examples/text_graph/erniesage/config/erniesage_link_prediction.yaml @@ -1,7 +1,7 @@ # Global Enviroment Settings # trainer config ------ -n_gpu: 2 # delete it, if use cpu to train +n_gpu: 1 # number of gpus used to train, delete it, if use cpu seed: 2020 task: "link_prediction" diff --git a/PaddleNLP/examples/text_graph/erniesage/data/dataset.py b/PaddleNLP/examples/text_graph/erniesage/data/dataset.py index 173af3db..91eeed1a 100644 --- a/PaddleNLP/examples/text_graph/erniesage/data/dataset.py +++ b/PaddleNLP/examples/text_graph/erniesage/data/dataset.py @@ -21,7 +21,7 @@ from paddle.io import Dataset, IterableDataset from paddlenlp.utils.log import logger import pgl from pgl import Graph -from pgl.nn.sampling import graphsage_sample +from pgl.sampling import graphsage_sample __all__ = [ "TrainData", -- GitLab