未验证 提交 69ccb4c4 编写于 作者: Z Zhong Hui 提交者: GitHub

fix for pgl for the erniesage

fix for pgl for the erniesage
上级 5b4aef8e
# Global Enviroment Settings # Global Enviroment Settings
# trainer config ------ # 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 seed: 2020
task: "link_prediction" task: "link_prediction"
......
...@@ -21,7 +21,7 @@ from paddle.io import Dataset, IterableDataset ...@@ -21,7 +21,7 @@ from paddle.io import Dataset, IterableDataset
from paddlenlp.utils.log import logger from paddlenlp.utils.log import logger
import pgl import pgl
from pgl import Graph from pgl import Graph
from pgl.nn.sampling import graphsage_sample from pgl.sampling import graphsage_sample
__all__ = [ __all__ = [
"TrainData", "TrainData",
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册